Ubuntu 18安装CUDA10.2

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说明就好了

以上