Ubuntu18.04安装nvidia驱动450、cuda10.0、cudnn7.6

标题Ubuntu18.04安装nvidia驱动、cuda10.0、cudnn

1. 卸载原驱动

nvidia-uninstall

2. 禁用nouveau驱动

2.1 打开blacklist.conf

vim /etc/modprobe.d/blacklist.conf

2.2 将nouveau驱动加入黑名单，在blacklist.conf末尾添加

blacklist nouveauoptions nouveau modeset=0

2.3 保存退出后，更新系统

 sudo update-initramfs -u

2.4 重启系统（必须）

reboot

3. 安装新驱动

3.1 验证nouveau是否已禁用

lsmod | grep nouveau

若无输出，则表明nouveau已禁用

3.2 关闭图形界面

sudo service lightdm stop

3.3 给驱动run文件赋予执行权限

chmod a+x NVIDIA-Linux-x86_64-450.66.run

3.4 安装驱动

./NVIDIA-Linux-x86_64-450.66.run -no-x-check -no-nouveau-check -no-opengl-files

说明：
-no-x-check：关闭X服务
-no-nouveau-check：禁用nouveau
-no-opengl-files：只安装驱动文件，不安装OPenGL文件

然后，按提示选择相应的选项回车即可

3.5 挂载nvidia驱动

modprobe nvidia

3.6 检查驱动是否安装成功

nvidia-smi

查看是否有显卡信息输出

4. 安装cuda

4.1 给驱动run文件赋予执行权限

chmod a+x cuda_10.0.130_410.48_linux.run

4.2 安装cuda

 ./cuda_10.0.130_410.48_linux.run

此时会出现一大堆提示信息，可按空格键跳过，直至出现安装选项，安装选项如下，可按如下输入yes/no。

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: no

Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: yes

Enter Toolkit Location
[ default is /usr/local/cuda-10.0 ]:

Do your want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: no

Install the CUDA-10.0 Samples?
(y)es/(n)o/(q)uit: no

4.3 添加环境变量

在/root/.bashrc中添加

export PATH="/usr/local/cuda-10.0/bin:$PATH"export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH"

4.4 检测cuda是否安装成功

nvcc -V

查看是否有cuda版本信息输出

5. 安装cudnn

5.1 解压tgz压缩包

tar -zxvf cudnn-10.0-linux-x-v7.6.4.38.tgz

得到cuda文件夹

5.2 将cuda文件夹下的文件拷贝至步骤4中相应安装目录

cp cuda/include/cudnn.h /usr/local/cuda-10.0/include/
cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64/

5.3 增加可读权限

sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*

6. 相关链接

nvidia驱动下载
cuda下载
cudnn下载
免注册下载cudnn方法