Cant get GPU support working on Windows 10

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


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

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

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