torch.cuda.is_available()返回False的解决办法

torch.cuda.is_available()返回False的解决办法

之前显卡驱动和cuda都可以正常使用，但是今天运行程序时发现使用了CPU，没有使用GPU，运行的时候会报错说 GPU驱动没有运行之类的错误，然后使用CPU进行了训练（直观感觉就是慢了很多）。

1.首先查看cuda 版本。

cat /usr/local/cuda/version.txt

//  输出正常：
 CUDA Version 11.0.228

2.查看是否能使用GPU。结果为False

python3
import torch
torch.cuda.is_available() # cuda是否可用
或print(torch.cuda.is_available())
torch.cuda.device_count() #返回gpu数量；

接着检查了驱动，有设备为0的GPU啊，但是不可用GPU，

怀疑出现系统更新造成的显卡驱动坏了，重新安装了显卡驱动。问题终得解决！！！

后记：

torch.cuda.device_count()
#返回gpu数量；

torch.cuda.get_device_name(0)
#返回gpu名字，设备索引默认从0开始；

torch.cuda.current_device()
#返回当前设备索引；