本文共 559 字，大约阅读时间需要 1 分钟。

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()

#返回当前设备索引；

转载地址：http://lhtki.baihongyu.com/