Torch.cuda.is_available() 不断切换到 False

Question

我尝试了几种解决方案，这些解决方案暗示了当 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/5 https://github.com/pytorch/pytorch/issues/15612

注意：当torch.cuda.is_available()工作正常但在某些时候切换到 时False，我必须重新启动计算机，然后它再次工作（一段时间）。

Answer 1

torch.cuda.is_available()造成这种情况的原因是和False的版本不兼容。pytorchcudatoolkit

截至 2022 年 6 月，当前版本的 pytorch 与cudatoolkit=11.3兼容，而当前的cuda 工具包版本 = 11.7。来源

解决方案：

0. 卸载 Pytorch 进行全新安装。如果不强制安装（使用pip install --upgrade --force-reinstall <package_name>.

1. 运行conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch安装pytorch。

2. 从https://developer.nvidia.com/cuda-11.3.0-download-archive安装 CUDA 11.3 版本。

你已准备好出发。

Answer 2

也曾torch.cuda.is_available ()有过假。
但是当安装Nvidia驱动到最新版本436.48时，却显示True。我之前将Pytorch 更新到 1.2.0。我有 Windows 10 和Anaconda。

Answer 3

1. 按照此链接中的说明，使用 apt-get 安装 CUDA 9.1：

https://cryptoandcoffee.com/mining-gems/cuda-9-0-install-ubuntu-16-04-apt-get/

2. 使用 pip 安装 PyTorch：

   pip install torchvision ( this will install both torch and torchvision )
   

3. 重新启动

现在尝试一下：

~$ python -c 'import torch; print torch.cuda.is_available()'