I am a bit new to torch and have only used it on VMs before. I am now trying to use it on my computer and it doesn’t seem to be working.
print(pycuda.driver.Device.count()) I get
1, however when I
print(torch.cuda.is_available()) I get
I tried reinstalling both torch and cuda and I still get this issue. Can anyone point me in the right direction?
You might have installed the CPU-only binary of PyTorch. Could you check what
That releases None. Could you point me to how to install a GPU version of PyTorch? I tried uninstalling and reinstalling a cuda version, but I’m unsure if I am entirely cleaning PyTorch.
Thanks for the update as it confirms you’ve installed the CPU-only binary.
Refer the these install instructions, select the desired CUDA runtime (11.7 or 11.8), copy/paste the install command, and make sure the PyTorch binary indicates the CUDA version in its name.
Also, I would recommend installing PyTorch into a clean and empty environment to avoid conflicts with already installed PyTorch versions.
2.0.0+cu117 a simple
pip install torch would work.
Sorry for the late reply, but that has worked. Thank you so much!