RuntimeError: No CUDA GPUs are available

问题描述

  1. 使用pytorch进行模型训练，结果发现没有可用的CUDA GPU，而实际电脑上是有的
  2. 尝试了网上常见方法，比如检测CUDA是否可用之类的方法均无果

问题解决

  1. 偶然间找到了这篇博客升级显卡驱动，尝试一番后刚好解决了我的问题，如果你在尝试了问题描述中的方法还未能解决你的问题，不妨像我一样升级一下显卡驱动程序，这并不能保证解决你的问题，不过也是一种值得尝试的方法
  2. 首先查看一下可用的显卡驱动，进行驱动程序查看以及安装
ubuntu-drivers devices

  1. 之后选择系统推荐的带有“non-free recommended”字样的版本进行安装，可以看到我的推荐是510版本
  2. 使用如下命令进行安装
sudo apt-get install nvidia-driver-510
  1. 安装成功后重启计算机就不会出现标题报错了

更多推荐

RuntimeError: No CUDA GPUs are available