I’ve been trying to install PyTorch with CUDA support enabled, but have been unsuccessful no matter what I try.
GPU: GTX 1080 Ti
OS: Windows 10
I installed Anaconda and ran the following command in the Anaconda Prompt:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch.
As far as I can tell, this is how I download the version of PyTorch with GPU support.
This is the output of
NVIDIA-SMI 496.49 Driver Version: 496.49 CUDA Version: 11.5
This is what happens when I try to access PyTorch from within the Python3 interpreter:
import torch torch.cuda.is_available() > False torch.__version__ > '1.10.0+cpu' torch.cuda.device_count() > 0 torch.cuda.init() > AssertionError: Torch not compiled with CUDA enabled
It appears that the CPU version is installed, even though I didn’t select the CPU version on the PyTorch website. What can I do to ensure I install the GPU version? Thanks for all your help!