Ubuntu18中安装Nvidia驱动和CUDA和cuDNN库加速
1 安装Nvidia驱动
主要分为两种方法,第一种是用Software&Updates切换对应Nvidia驱动,第二种是官网下载驱动文件然后安装。(注:请关闭电脑的security mode)
1. Software&Updates 中切换驱动
打开Software & Updates,选择 Additional Drivers。切换到对应的显卡驱动版本,然后重启即可
$ nvidia-smi
查看
若要卸载,切换回自带驱动即可
1.2 官网下载驱动文件然后安装
禁用nouveau
$ sudo vim /etc/modprobe.d/blacklist.conf
末尾添加
blacklist nouveau
options nouveau modeset=0
验证nouveau是否已经被禁用
lsmod | grep nouveau
没有信息显示,说明nouveau已被禁用,接下来可以安装nvidia显卡驱动了
下载显卡驱动
在英伟达的官网上查找你自己电脑的显卡型号然后下载相应的驱动下载后的run文件拷贝至home目录下。
安装
在ubuntu下按ctrl+alt+f1进入命令行界面,然后在命令行界面下输入:
$ sudo service lightdm stop //这个是关闭图形界面,不执行会出错。
然后卸载掉原有驱动:
$ sudo apt-get remove nvidia-* (若安装过其他版本或其他方式安装过驱动执行此项)
给驱动run文件赋予执行权限:
$ sudo chmod a+x NVIDIA-Linux-x86_64-xxxxxxx.run
安装驱动文件
$ sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files
-no-x-check:安装驱动时关闭X服务
-no-nouveau-check:安装驱动时禁用nouveau
-no-opengl-files:只安装驱动文件,不安装OpenGL文件
安装过程选项:
The distribution-provided pre-install script failed! Are you sure you want to continue? 选择 yes 继续。
Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 选择 No 继续。
问题没记住,选项是:install without signing
问题大概是:Nvidia’s 32-bit compatibility libraries? 网上说选择 No 继续,这里只出现了yes选项,但后面也没发现问题。
Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择 Yes 继续
验证
查看显卡驱动相关的信息,能显示即安装成功
$ nvidia-smi
重启,桌面分辨率正常即可
2 安装CUDA和cuDNN
2.1 安装CUDA
进入nvidia开发者网站的CUDA下载页面
CADA Toolkit Archive
选择runfile格式的CUDA文件下载
下载完成后,解压,并运行上图中的命令,会有条款,接受即可,注意安装CUDA的时候不要安装驱动(因为在第一步我们已经安装过了)。
然后,
$ sudo gedit ~/.bashrc
在文件最后一行添加:
export PATH=/usr/local/cuda-11.1/bin:/usr/local/cuda-11.1/NsightCompute-2020.2.0${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
最后,使配置生效
$ source ~/.bashrc
进入/usr/local/cuda-11.1/samples/1_Utilities/目录
$ cd deviceQuery
$ sudo make
$ ./deviceQuery
如下输出,则CUDA安装成功
2.2 安装cuDNN
进入下载页面
NVIDIA cuDNN
选择合适的版本下载(cuDNN Library for Linux),
然后解压,并进入到相应目录,运行以下命令:
$ sudo cp cuda/include/cudnn.h /usr/local/cuda-11.1/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64
$ sudo chmod a+r /usr/local/cuda-11.1/include/cudnn.h
$ sudo chmod a+r /usr/local/cuda-11.1/lib64/libcudnn*
以配置cuDNN环境。
通过
$ cat /usr/local/cuda-11.1/include/cudnn.h | grep CUDNN_MAJOR -A 2
查看cuDNN版本
(若无输出,可查看此链接)
3 显卡算力查看
查看此链接