Not relevant to PyTorch
Have a look at
  
      

      NVIDIA Developer – 19 Nov 13
  

  
    

GPU Accelerated Computing with Python 8

  GPU-Accelerated Computing with Python NVIDIA’s CUDA Python provides a driver and runtime API for existing toolkits and libraries to simplify GPU-based accelerated processing. Python is one of the most popular programming languages for science,...