PyTorch + CUDA 11.4

Yes, you can build PyTorch from source using all released CUDA versions between 10.2 and 11.5.

2 Likes