Does the official conda install include / require cudnn?

deployment
1

I notice that the official conda install doesn’t include cudnn. Is that required separately? Or it is included in the cudatoolkit?

FYI the website says to use:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

2

cuDNN is statically linked into PyTorch, as no conda binaries or pip wheels are available for it.

3

so that mean I need to manually make sure they are available or all it is all good?

If that conda command runs successfully and I torch is showing cuda capabilities then all good?

4

Yes, the binaries will ship with cuDNN and you can check the version via print(torch.backends.cudnn.version()).

1 Like
5

Cheers! Thanks for clarifying