超详细教程——Ubuntu20.04 安装英伟达NVIDIA显卡驱动、CUDA、Cmake以及不同版本的CUDA切换

在windows和linux双系统下,为ubuntu安装nvidia、cuda、cmake等程序是一个可能会难到各个计算机小白的问题,本文将一步步地带您学会安装以上内容。


一、安装英伟达NVIDIA显卡驱动

对于新安装的linux操作系统,我们需要为它安装nvidia显卡驱动(ubuntu有自带的显卡驱动但更推荐安装nvidia)

看了很多网上安装nvidia的教程,但大多数是都是很复杂繁琐的,对于新手而言是极其不友好的

在这里介绍一种有用、快速的安装方法:

1.进入linux系统,如图,找到应用”软件和更新“

 打开它,点击“附加驱动”

 第一次打开时,ubuntu默认使用驱动为X.Org X server - Nouveau display driver

我们需要把它禁用,具体操作为选择其他驱动(列表上你所需要的驱动),如这里我们选择的是

“NVIDIA driver metapackage 来自 nvidia -driver-510-server”

 然后点击“应用更改”,输入密码认证后,等待进度条完成后,重启系统,即可完成安装。

如果遇到configurling secure boot 的弹窗,则根据提示,新设置一个验证密码(并非系统的等入密码),后输入指令$ reboot 重启电脑。

 在未完全开机时会,会显示出:

Continue boot

Enroll MDK

Enroll key from disk

Enroll hash from disk

这时我们只需要选择第二个选项Enroll MDK后,输入刚才新设置的密码(输入的密码不会显示出来),完成后会发现只剩下其他三个选项,然后我们再选择第一个Continue boot继续开机即可完成安装。

待系统重启完成后,输入以下命令,验证安装:

$ nvidia-smi

看到终端:

 后安装成功。


二、安装Cmake

这里只介绍用sudo apt来安装cmake

优点:方便卸载,流程短;缺点:无法安装指定版本的cmake。

输入如下指令即可完成cmake安装:

$ sudo apt update
$ sudo apt full-upgrade -y
$ sudo apt install cmake

 验证安装:

$ cmake --version

看到输出如图后,cmake安装成功

   


三、安装CUDA  

输入nvidia-smi命令查看支持的cuda版本

我们可以看到GPU支持的cuda的最高版本为11.6(注意一些版本的cuda与gcc/g++的版本存在不兼容问题)

在https://developer.nvidia.com/cuda-toolkit-archive找到你需要的cuda版本,并下载

这里以11.6为例

 

 

 根据官方安装教程,选择runfile(local)版的,输入指令:

wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda_11.6.0_510.39.01_linux.run
sudo sh cuda_11.6.0_510.39.01_linux.run

等待运行(可能会有点慢),进入页面

选择Continue,继续

进入下一个页面,输入accept

进入下一个页面,将第一个安装选项取消后点击install,即可完成安装

 最后配置环境变量,输入指令:

$ gedit ~/.bashrc

打开文件,在文件末尾输入以下语句,保存

export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

如下:

 

 更新环境变量配置,输入:

$ source ~/.bashrc

安装完成

验证安装:

$ nvcc -V

 

安装成功(这里由于我电脑用的是10.0版本故显示为10.0)


四、切换不同版本的CUDA

安装其他版本的cuda,过程如上,但在需要切换使用的cuda版本时,需要对环境变量进行修改

具体操作为:

输入

$ export PATH=/usr/local/cuda-xx.x/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda-xx.x/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
$ source ~/.bashrc

即可更改cuda版本,输入完成后用nvcc -V查看版本是否所需要的版本。