1.情况:
输入nvidia-smi报错:
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.
以及输入torch.cuda.is_available()返回false
2.解决:
只需执行两条命令就好:
sudo apt-get install dkms
sudo dkms install -m nvidia -v 470.94(470.94表示的是驱动版本号)
利用命令 ll /usr/src/ 可查看下面有一个nvidia-470.94/文件夹,版本号因电脑而异