Hi, I’ve the following issue in my program. NVIDIA GeForce RTX 3080 with driver version 516.94 have been installed on my PC.
I tried uninstall and reinstall cudatookit, pytorch, and reinstalling. torch.cuda.is_available() is still False though torch.version.cuda can check as
Any guidance would be warmly welcome and thanks in advance.
Your system is most likely not able to communicate with the driver, which could happen e.g. if you didn’t restart the machine after a driver update.
Make sure other CUDA samples are running first, then check PyTorch again.
Thanks for your suggestion. Now the CUDA is available but I’d like to know why CUDA device count 1 is displayed in red color?
That’s your IDE’s color scheme and isn’t defined by PyTorch.
Hey @naychelynn , how did you solve the error? I am using GeForce RTX 3090.
I exactly follow @ptrblck 's suggestion “Restart my PC”.
I’ve tested on other environments too and it’s the same. ( installing torch without restarting then the same problem and CUDA is available after restart process).