What is the minimum cuda environment for pytorch

deployment
1

pytorch1.7.1+cuda11.0

  1. If I want to run pytorch with CUDA, what is the minimum cuda environment? (C++/CUDA extension is not required). Should I install all CUDA libraries? (cuNPP, cuSPARSE, cuBLAS, NCCL etc.)
  2. And, Does the torch-1.7.1+cu110-cp36-cp36m-linux_x86_64.whl contains a cuDNN lib? Should I install an additional package If I want to use cuDNN in pytorch?
2

You would only need to install an NVIDIA driver to run PyTorch if you install the pip wheels or conda binaries, as they ship with the CUDA runtime, cuDNN, NCCL, cublas, etc.

1 Like