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()
#返回当前设备索引；
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/99872.html
