ubuntu20.04算法服务器cuda 环境搭建

cuda 版本：10.1
nvidia驱动版本：Driver Version: 440.118.02
系统环境：ubuntu20.04
kernel: 5.4.0-62-generic

参考文档：
cuda10.1官方安装指导
Ubuntu16.04+cuda10.1官方安装指导的补充说明
Ubuntu 20.04 CUDA&cuDNN安装方法

一、安装前准备

在Linux上安装CUDA工具包和驱动程序之前，必须采取一些行动:

1、验证系统是否有支持cuda的GPU。

algorithm_team_admin@algorithmserver:~$ lspci | grep -i nvidia
01:00.0 VGA compatible controller: NVIDIA Corporation TU102 [GeForce RTX 2080 Ti] (rev a1)
01:00.1 Audio device: NVIDIA Corporation TU102 High Definition Audio Controller (rev a1)
01:00.2 USB controller: NVIDIA Corporation TU102 USB 3.1 Host Controller (rev a1)
01:00.3 Serial bus controller [0c80]: NVIDIA Corporation TU102 USB Type-C UCSI Controller (rev a1)
02:00.0 VGA compatible controller: NVIDIA Corporation TU102 [GeForce RTX 2080 Ti] (rev a1)
02:00.1 Audio device: NVIDIA Corporation TU102 High Definition Audio Controller (rev a1)
02:00.2 USB controller: NVIDIA Corporation TU102 USB 3.1 Host Controller (rev a1)
02:00.3 Serial bus controller [0c80]: NVIDIA Corporation TU102 USB Type-C UCSI Controller (rev a1)

2、验证系统运行的是支持的Linux版本。

algorithm_team_admin@algorithmserver:~$ uname -m && cat /etc/*release
x86_64
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=20.04
DISTRIB_CODENAME=focal
DISTRIB_DESCRIPTION="Ubuntu 20.04 LTS"
NAME="Ubuntu"
VERSION="20.04 LTS (Focal Fossa)"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 20.04 LTS"
VERSION_ID="20.04"
HOME_URL="https://www.ubuntu.com/"
SUPPORT_URL="https://help.ubuntu.com/"
BUG_REPORT_URL="https://bugs.launchpad.net/ubuntu/"
PRIVACY_POLICY_URL="https://www.ubuntu.com/legal/terms-and-policies/privacy-policy"
VERSION_CODENAME=focal
UBUNTU_CODENAME=focal

3、检查系统是否安装了gcc。

因为Ubuntu20.04自带的gcc版本为9.3，而cuda10.1不支持gcc-9，因此要手动安装gcc-7，命令如下：

#关键点：gcc降级
#因为Ubuntu20.04自带的gcc版本为9.3，而cuda10.1不支持gcc-9，因此要手动安装gcc-7，命令如下：sudo apt-get install gcc-7 g++-7
#安装完gcc-7，系统中就存在两个版本的gcc，因此要设置默认的gcc，命令如下：
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
#此命令可以通过update-alternatives设置gcc各版本的优先级，优先级最高的为系统默认版本，可以用下述命令显示其优先级：
sudo update-alternatives --display gcc
#设置默认的g++也是如此：
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 9
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 1
#显示g++优先级：
sudo update-alternatives --display g++

查看当前gcc版本
algorithm_team_admin@algorithmserver:~$ gcc --version
gcc (Ubuntu 7.5.0-6ubuntu2) 7.5.0
Copyright (C) 2017 Free Software Foundation, Inc.
This is free software; see the source for copying conditions. There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

4、检查系统是否安装了正确的内核头文件和开发包。

#运行以下命令可以找到您的系统正在运行的内核版本:
uname -r
# 当前运行的内核可以使用以下方式安装内核头文件和开发包:
sudo apt-get install linux-headers-$(uname -r)
 
 
#查看Linux中安装了哪些内核：
dpkg --get-selections | grep linux  或者 dpkg --list |grep linux

5、禁用Nouveaunon

# 1、如果下面的命令打印了任何东西，就会加载Nouveaunon驱动程序:
$ lsmod | grep nouveau
# 2、若上述命令有输出，则创建文件 /etc/modprobe.d/blacklist-nouveau.conf 包含以下内容
blacklist nouveau
options nouveau modeset=0
# 3、重新生成内核initramfs:
algorithm_team_admin@algorithmserver:~$ sudo update-initramfs -u