ubuntu19.10+nvidia-440+cuda-10.2+cudnn-7.7安装

该方案为笔者尝试安装正确后的经验总结,参考文档:

  1. Ubuntu安装和卸载CUDA和CUDNN
  2. Ubuntu安装gcc g++ 7
  3. NVIDIA CUDA Installation Guide for Linux
  4. Ubuntu 18.04 gcc降级为4.8版本
  5. 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

安装完成。

你可能感兴趣的:(基础环境)