CUDA已经更新到10.2了,并且支持的最低的显卡驱动版本是440。然鹅...系统自带的显卡最高版本是435,要自己安装最新版本的显卡驱动,这就有点烦。
第一步,禁用nouveau
这里网上教程一搜一大把,汇总一下,首先在命令行中敲入,
sudo vim /etc/modprobe.d/blacklist-noureau.conf
然后写入
blacklist nouveau
options nouveau modeset=0
之后,在命令行中敲入
sudo update-initramfs -u
重启,然后
lsmod | grep nouveau
确认一下是否已经禁用
第二步,安装显卡驱动以及CUDA10.2
其实现在不用单独安装显卡驱动了,在cuda_10.2.89_440.33.01_linux.run这个文件里面,已经帮你内置了440的驱动,直接安装就好。一路按照提示装下去。。
运行程序,
sudo sh cuda_10.2.89_440.33.01_linux.run
之后可以校验一下有没有装好。
第三步,安装CUDNN
注册账号之后,下载文件,然后
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
到这里安装就已经好了。想要检验一下有没有好?
安装pytorch, 然后运行一下
import torch
print(torch.cuda.is_available())
如果看到True说明就好了
以上