问题和现象
使用 conda 安装 pytorch-gpu 之前,使用 nvidia-smi
确认过显卡驱动和 CUDA 版本是正常的。使用 conda 安装 pytorch-gpu 之后,import torch
可以正常工作,但是
torch.cuda.is_available()
返回 False
,nvidia-smi
亦现实显卡驱动错误。Ubuntu 软件与更新中显示,显卡驱动被配置为 noveau
可能是由于 conda 安装 pytorch-gpu 的过程破坏了显卡驱动的状态
解决
在 Ubuntu 软件与更新界面中(由 Ubuntu Software)进入,将显卡驱动配置为 nvidia ,重启计算机。