该方案为笔者尝试安装正确后的经验总结,参考文档:
- Ubuntu安装和卸载CUDA和CUDNN
- Ubuntu安装gcc g++ 7
- NVIDIA CUDA Installation Guide for Linux
- Ubuntu 18.04 gcc降级为4.8版本
- ubuntu18.04 + CUDA10 + CUDNN7 deb安装
一、安装包准备
首先下载所需安装包,驱动、CUDA-10.2、CUDNN7_7
1.下载驱动
驱动官网下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn ,根据自己显卡的情况下载对应版本的显卡驱动,比如笔者的显卡是RTX2060 SUPER
下载成功后得到一个后缀为.run的安装包:
NVIDIA-Linux-x86_64-440.36.run
2.下载CUDA
我们可以在官网:CUDA10下载页面根据相应版本进行下载,笔者下载的是:
cuda_10.2.89_440.33.01_linux.run
3. 下载CUDNN
进入CUDNN官网,然后点击Download开始选择下载版本,当然在下载之前还有登录,选择版本界面如下,我们选择cuDNN Library for Linux,笔者下载的是cuDNN Runtime Library for Ubuntu18.04 (Deb)的文件,这个比压缩包安装的方式更快捷:
libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
二、安装显卡驱动
1. 禁用nouveau驱动
sudo vim /etc/modprobe.d/blacklist.conf
在文本后添加:
blacklist nouveau
options nouveau modeset=0
然后执行:
sudo update-initramfs -u
重启后,执行以下命令,如果没有屏幕输出,说明禁用nouveau成功:
lsmod | grep nouveau
2. 安装新驱动
切换到命令行终端:
Ctrl-Alt+F3
执行以下命令禁用X-Window服务,否则无法安装显卡驱动:
sudo service lightdm stop
安装新驱动:
sudo chmod +x NVIDIA-Linux-x86_64-440.36.run
sudo ./NVIDIA-Linux-x86_64-440.36.run
执行以下命令启动X-Window服务:
sudo service lightdm start
最后执行重启命令,重启系统即可:
reboot
注意: 如果系统重启之后出现重复登录的情况,多数情况下都是安装了错误版本的显卡驱动。需要下载对应本身机器安装的显卡版本。
三、安装CUDA-10.2和CUDNN-7_7
由于CUDA-10.2支持的gcc版本为7.3,而ubuntu-19.10自带的gcc/g++版本为9.2,故需要安装gcc-7.3。
1. 安装gcc和g++,从9.2降级到7.3或者7.4
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
sudo apt-get install gcc-7
sudo apt-get install g++-7
把gcc g++ 的连接指向gcc-7 g+±7
$ cd /usr/bin
$ sudo rm gcc
$ sudo ln -s gcc-7 gcc
$ sudo rm g++
$ sudo ln -s g++-7 g++
2. 安装CUDA
chmod +x cuda_10.2.89_440.33.01_linux.run
sudo ./cuda_10.2.89_440.33.01_linux.run
注意:在安装时,会弹出安装组件的界面,将第一个选项驱动安装取消掉,否则会安装失败
安装完成之后,可以配置他们的环境变量,在vim ~/.bashrc的最后加上以下配置信息:
export CUDA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
最后使用命令source ~/.bashrc使它生效。
可以使用命令nvcc -V查看安装的版本信息。
测试是否安装成功,执行以下几条命令:
cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
正常情况下会输出GPU信息。
3. 安装CUDNN
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
安装完成。