我已经尝试了几种解决方案,这些解决方案暗示了当CUDA GPU可用并且安装了CUDA但Torch.cuda.is_available()
返回False
时该怎么做。它们确实有帮助,但只是暂时的,这意味着torch.cuda-is_available()
报告了True,但过了一段时间,它又切换回False。我使用CUDA 9.0.176和GTX 1080。我该怎么做才能得到永久的效果?
我尝试了以下方法:
https://forums.fast.ai/t/torch-cuda-is-available-returns-false/16721/5https://github.com/pytorch/pytorch/issues/15612
注意:当torch.cuda.is_available()
工作正常,但在某个时候切换到False
,那么我必须重新启动计算机,然后它再次工作(一段时间)。
5条答案
按热度按时间ajsxfq5m1#
torch.cuda.is_available()
导致False
的原因是pytorch
和cudatoolkit
版本之间的不兼容。截至2022年6月,当前版本的pytorch与cudatoolkit=11.3兼容,而当前的cuda工具包版本= 11.7。Source
解决方案:
1.卸载Pytorch进行全新安装。如果不强制安装(使用
pip install --upgrade --force-reinstall <package_name>
),则无法在新版本之上安装旧版本。1.运行
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
安装pytorch。1.从https://developer.nvidia.com/cuda-11.3.0-download-archive安装CUDA 11.3版本。
你可以走了
h9vpoimq2#
torch.cuda.is_available ()
也有false。但将Nvidia驱动安装到最新版本436.48时,显示True。我之前更新了Pytorch到1.2.0。Windows 10和Anaconda
oyxsuwqo3#
1.使用apt-get安装CUDA 9.1,请按照此链接中的说明操作:
https://cryptoandcoffee.com/mining-gems/cuda-9-0-install-ubuntu-16-04-apt-get/
1.使用pip安装PyTorch:
1.重启
现在试试:
hs1rzwqc4#
我也看到了这个问题。原因是Pytorch使用的CUDA版本与安装的Nvidia驱动程序不同步。正如Joe的回答一样,解决方案是更新Nvidia驱动程序。其他一些需要注意的重要背景信息:
nvidia-smi
检查您的Nvidia驱动程序版本。*Pytorch预装了一个CUDA版本,可能与您计算机上安装的版本不同。
nvidia-smi
时显示的版本。即使您的驱动程序版本与此CUDA版本兼容,也可能与Pytorch CUDA版本不兼容。torch.version.cuda
变量来获取Pytorch CUDA版本。这是确定所需Nvidia驱动程序版本的版本。xghobddn5#
我的解决方案是使用pip而不是conda重新安装pytorch
而不是
(Don不介意上面的 Torch 版本差异;这只是为了说明。)