Will "conda install pytorch torchvision -c pytorch" also install CUDA and cuDNN?


So as far as I understand CUDA and cuDNN are automatically installed when running pip/pip3 install torch. But what about when using conda install pytorch torchvision -c pytorch? Does the PyTorch Conda package also come with it’s own copies of CUDA and cuDNN?


As far as I know, yes, the conda package also ships with its CUDA and cuDNN binaries.
This build script also suggests the same.


I can confirm ptrblck’s answer. pytorch 0.4 statically links to cudnn, nccl and most cuda libraries within itself.

(ngimel) #4

Almost :slight_smile: pytorch conda package still requires conda cuda toolkit package, which will bring the following unnecessary goodies with it: