ubuntu20.04安装NVIDIA显卡驱动+CUDA

1 安装驱动

1.1 查看显卡的型号

打开终端，输入指令以查看电脑的显卡型号：·

1	lspci \| grep -i vga

ubuntu-drivers devices

可以看到如下界面：

model即为显卡的型号信息，此处为GeForce RTX 2070 SUPER；推荐的显卡驱动版本号为nvidia-driver-450 - distro non-free。

1.2 官网下载显卡驱动

到官网下载对应的显卡驱动：

下载最新版官方 GeForce 驱动程序www.nvidia.cn

ubuntu系统的话，选择对应显卡型号的Display Driver进行下载。

1.3 安装相关依赖

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

1.4 禁用系统默认显卡驱动

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

打开文件，在文件末尾写入：

1 2	blacklist nouveau options nouveau modeset=0

保存后手动更新；

1	sudo update-initramfs -u

电脑重启，输入下列指令进行确认，若无输出，则禁用成功：

1	lsmod \| `grep` nouveau

1.5 停止lightdm桌面服务

1	sudo service lightdm stop

如果提示没有lightdm，就安装一下，再执行上面的命令

1	sudo apt install -y lightdm

（因为后面使用的命令行界面tty1~7 需要lightdm）
接下来按住键盘的Ctrl+Alt+F1，如果黑屏就切换到tty2，按Ctrl+Alt+F2
如果Ctrl+Alt+F1后还显示图形化桌面就重启一下电脑，之后不要登录，在登录界面直接按Ctrl+Alt+F1或F2，这时应该进了命令行界面了，然后登录root用户或者你的普通用户，如果登录root用户时所有密码都登录失败，一般是因为你装系统时没设置root密码的原因，这时就需要设置一下root账户密码，执行以下命令：

1	sudo passwd root

然后会让你输入你的普通用户密码，再输入两次root密码，然后就可以在命令行界面登录root账户了。

1.6 安装驱动

（1）首先查看你有没有安装gcc

1	gcc --version

如果没有安装gcc一般make也没安装，这时需要安装一下gcc和make

sudo apt install gcc 
sudo apt install make
# 或者直接执行
sudo apt install gcc & make # 同时安装gcc和make，不用一条一条执行了，效果和上面两条命令相同

（2）执行安装
cd到你下载的显卡驱动的目录
执行：

sudo chmod a+x NVIDIA-Linux-x86_64-450.80.02.run
sudo ./NVIDIA-Linux-x86_64-450.80.02.run -no-x-check -no-nouveau-check -no-opengl-files
# -no-x-check:安装时关闭X服务
# -no-nouveau-check: 安装时禁用nouveau
# -no-opengl-files:只安装驱动文件，不安装OpenGL文件

下面者两个按图中选择，其他默认就好

安装完成以后执行

1	nvidia-smi

如果显示如下内容则表示安装成功

2 安装CUDA

从上面的图中可以看到我的显卡最高可以支持cuda11.0(绿色框内)
下载cuda https://developer.nvidia.com/cuda-toolkit-archive
首先需要知道不同版本的cuda需要的gcc版本是不同的，cuda11.0与gcc的对应关系如下：

其他cuda版本与gcc的版本关系可以在cuda对应版本后面的document—>Installation Guide Linux下找到，不再赘述。

找到自己需要的cuda版本如下选择，最下面虚线框内会给出下载和命令

在终端中执行：

1	wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run

下载完成后进行cuda安装
cd到下载的cuda所在目录

1 2	# 安装cuda sudo sh cuda_11.0.2_450.51.05_linux.run

正常安装的情况下，会出现下面显示：

这是软件协议，跟普通软件安装时一样，这时输入： accept

在Driver处敲回车，选择不安装驱动，因为之前已经安装过驱动程序，这里是因为每个cuda都会自带一套符合当前版本cuda最低要求的驱动程序，如果这里选择安装的驱动的话，在Windows上安装时会将之前安装的驱动覆盖，但是在linux上覆盖安装的话，可能会出现错误，具体没有试过，如果想试试可以自行测试。我们使用之前自己下载的驱动。

然后按方向键将光标选中最后的Install再回车，开始安装，过一会儿会显示如下：

如果显示如上图，则表示安装完成，这时还不算安装成功，根据上面提示需要配置环境变量，进行如下操作：

# 打开.bashrc文件
gedit ~/.bashrc
# 将下面的11.0替换为你的cuda版本，其他不变，如果不知道自己安装的是哪个版本，就去/usr/local/文件夹下找一下
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

完事儿以后，执行如下命令更新变量，使其生效

1	source ~/.bashrc

在终端输入

nvcc -V

如果显示

至此，cuda的安装就算完成了。

3 参考

Ubuntu20.04安装NVIDIA显卡驱动+cuda+cudnn配置深度学习环境 (mlzhilu.com)