- I have NVIDIA GeForce GTX 950M
- I have installed CUDA 12 (cannot install older ones)
- I have installed cuDNN
- I have PyTorch 1.13.1+cpu
-
- “I ran pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116”
You’ve installed the CPU-only binaries, which is why torch.cuda.is_available()
returns False
.
" With CUDA
To install PyTorch via pip, and do have a CUDA-capable system, in the above selector, choose OS: Windows, Package: Pip and the CUDA version suited to your machine. Often, the latest CUDA version is better. Then, run the command that is presented to you"