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
Environment: Anaconda
Python: 3.9.8
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:
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!