Conda did not install cudnn for pytorch?

1

Hi,
I use the following command to install pytorch:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

then I use the command conda list to list the packages that have been installed with pytorch, I found the cudatoolkit but I did not found the cudnn, why? the pytorch do not need the cudnn ?
when I install tensorflow gpu version using the conda install command , the cudnn and cudatoolkit are both installed. Why pytorch only need cudatoolkit?

1 Like
2

pytorch comes with it’s own cudnn, so you don’t have to install it specifically.

# Use this to check if pytorch is using cudnn
torch.backends.cudnn.enabled
4 Likes
3

Got it，thank you very much！