Would PyTorch (for CUDA 12.1) work with CUDA 12.2

never used PTorch before and my internet connection is too limited to try cuda 12.2 myself.
should i download cuda 12.1 or cuda 12.2 for it?
and if it doesn’t, when do you expect PyTorch will support CUDA 12.2?

PyTorch binaries ship with their own CUDA dependencies and your locally installed CUDA toolkit won’t be used unless you build PyTorch from source or a custom CUDA extension.

1 Like

Alright. By the way do you know if i need to install cuDNN seperately or does it also come with PyTorch?

cuDNN also comes with PyTorch as well as NCCL, cuSPARSE, cuSOLVER, cuBLAS, etc.

1 Like

i’ll delete my own versions of cudnn and cuda then (since i only planned to use them with pytorch)
thanks alot!