CUDA 12.2.1 and PyTorch

Hello, I have CUDA 12.2.1 installed on my computer. Will this CUDA version be compatible with the torch, torchvision, and torchaudio libraries that are compatible with CUDA 12.1 as listed on PyTorch’s official website? Thank you very much.

Yes, since your locally CUDA toolkit won’t be used unless you build PyTorch from source or a custom CUDA extension.