Cant get GPU support working on Windows 10

1

I cant get my GPU working with Pytorch :worried:

  • clean Notebook, fresh Windows 10 installation.
  • newest Nvidia drivers, CUDA is installed and the demos are working fine.
  • Pytorch is installed as per instruction from the official webpage: conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

Still:

>>> import torch
>>> print(torch.version.cuda)
None

Do I have to specifically tell Python to use the CUDA version of Pytorch? What am I missing?

image2
image21045×614 35.9 KB

2

What does print(torch.version.cuda) return in your environment?