Pytorch Installation CUDA 11.4

Yes, since the PyTorch binaries ship with their own CUDA 11.8 runtime and would use the NVIDIA driver from your system which should be compatible with 11.8.

2 Likes