Does the official conda install include / require cudnn?

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

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

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?

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

1 Like

Cheers! Thanks for clarifying