Would PyTorch (for CUDA 12.1) work with CUDA 12.2

windows
1

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?

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
3

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

4

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

1 Like
5

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