I have a GTX 1080 on Windows 10 with the latest drivers and CUDA 10. After installing pytorch via Anaconda, I try to execute torch.cuda.is_available() in the python console but I get the above error. Anybody has run into a similar problem or knows what to do? I’ve reinstalled the graphics drivers, CUDA and the Python packages countless times, to no avail.
numba -s does list that it has detected the GPU and CUDA so not sure what’s wrong:
__CUDA Information__
Found 1 CUDA devices
id 0 b'GeForce GTX 1080' [SUPPORTED]
compute capability: 6.1
pci device id: 0
pci bus id: 1
Summary:
1/1 devices are supported
CUDA driver version : 10010
I’m sorry it is not an answer… I just wanted to mention that I am getting exactly the same error. Just that in my case I used Miniconda, and created a new environment for PyTorch. and installed the latest PyTorch in it. [conda install pytorch torchvision cudatoolkit=10.0 -c pytorch]. My env is the following:
I have the exact same issue on Windows with cuda 10 and the latest drivers. I believe this is an issue with pytorch and not CUDA. The other GPU enabled packages in my environment (catboost, numba) work fine. I fixed the issue by uninstalling NVIDIA drivers and miniconda, reverting to a previous system state (using system restore), and reinstalling everything again.
I installed it on the current version of miniconda3, which is available on their website. All my package version correspond to whatever’s available on the current version (without doing any updates). If I updated using conda update --all, the problem comes back. The only solution for now (until pytorch fixes this) is to install pytorch in a new environment.