print(torch.cuda.is_available())返回false的解决方法(未解决)

查看cuda版本

nvcc -V

print(torch.cuda.is_available())返回false的解决方法(未解决)_第1张图片
我的版本是10.1

查看驱动版本

nvidia-smi

print(torch.cuda.is_available())返回false的解决方法(未解决)_第2张图片

Command 'nvidia-smi' not found, but can be installed with:

sudo apt install nvidia-340               # version 340.108-0ubuntu5.20.04.1, or
sudo apt install nvidia-utils-390         # version 390.143-0ubuntu0.20.04.1
sudo apt install nvidia-utils-450         # version 450.119.03-0ubuntu0.20.04.1
sudo apt install nvidia-utils-450-server  # version 450.119.03-0ubuntu0.20.04.1
sudo apt install nvidia-utils-460         # version 460.73.01-0ubuntu0.20.04.1
sudo apt install nvidia-utils-435         # version 435.21-0ubuntu7
sudo apt install nvidia-utils-440         # version 440.82+really.440.64-0ubuntu6
sudo apt install nvidia-utils-418-server  # version 418.197.02-0ubuntu0.20.04.1
sudo apt install nvidia-utils-460-server  # version 460.73.01-0ubuntu0.20.04.1

好吧我没有安装驱动
我网上查到的是:
cuda的版本不能超过驱动对应的最高cuda版本。
我这里选择了460版本安装

sudo apt install nvidia-utils-460 

卸载不要的软件

sudo apt autoremove

在这里插入图片描述
NVIDIA-SMI失败,因为它无法与NVIDIA驱动程序通信。确保已安装并运行最新的NVIDIA驱动程序。
可能是没有重启的原因,重启完神奇的问题增加了
print(torch.cuda.is_available())返回false的解决方法(未解决)_第3张图片
感觉不太对劲,让我冷静一下

你可能感兴趣的:(linux下ubuntu,ubuntu,linux)