Could you create a new virtual environment and reinstall PyTorch?
The error message suggests that the currently found installation doesn’t ship with the CUDA runtime, so you might have multiple (old) installs in the current env.