真·手把手教你配置ubuntu20.04中cuda+cudnn

Posted 2021-10-25 真不错鸭

真·手把手教你配置ubuntu20.04中cuda+cudnn

• • 禁用自带显卡驱动
  • 安装显卡
  • 开始安装cuda
  • 安装cudnn
  • 多个版本的cuda

禁用自带显卡驱动

1.首先更新Ubuntu必备环境源

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential

2.执行下列代码打开ubuntu环境变量

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

3.在最后一行添加下面语句，保存即可

blacklist nouveau
options nouveau modeset=0

4.重启电脑

sudo reboot

5.重启之后输入下列命令，终端无其他显示代表成功

lsmod | grep nouveau

如图所示

安装显卡

1.进入链接: 英伟达显卡驱动下载页面.找到自己电脑合适的显卡驱动下载
ps:如果是英伟达30系或20系显卡选择GeForce，如果是笔记本选择后面带notebooks的即可

2.安装显卡依赖源

sudo apt-get install build-essential
sudo apt-get install libssl-dev
sudo apt-get install cmake

3.在终端输入"sudo sh+文件名" 安装显卡驱动文件，安装过程中选择第一个默认安装即可
例如：

sudo sh NVIDIA-Linux-x86_64-470.63.01.run

4.测试显卡驱动是否安装成功

nvidia-smi

开始安装cuda

1.调整Ubuntu的gcc、g++版本（按照cuda版本对应安装）
例如：安装cuda10.2版本需要gcc-7 g+±7，安装cuda11.1版本需要gcc-9 g+±9

apt-get install gcc-7 g++-7

切换多个版本的gcc g++

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100                  #
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50     

查看切换成功

sudo update-alternatives --config gcc

按数字键0 , 1 , 2切换

2.下载显卡对应cuda

进入英伟达官网下载链接如下：点击cuda下载

在终端输入，开始下载

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

下载完成后终端输入，开始安装

sudo sh cuda_10.2.89_440.33.01_linux.run

点击 continue 按钮，在对话栏最下方输入 accept 同意安装

重要ps:安装时把第一个选择显卡驱动安装选项点掉，在上述操作中已经安装好了显卡，不需要再次安装

选择 install 安装即可

3.安装完成后，给cuda添加环境变量，打开环境变量

gedit ~/.bashrc  

在最后一行输入（这里cuda-10.2应该改成cuda的安装版本）

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

最后更新环境变量即可

source ~/.bashrc

cuda安装完成！！！

测试代码

nvcc -V

安装cudnn

1.进入英伟达官网下载点击下载cudnn
ps:这里需要注册账号密码（建议QQ注册比较方便）

进入之后可以选择对应cuda版本进行下载cudnn

2.下载成功后解压文件，并在当前文件夹下输入命令
（将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹，将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中，并添加读取权限）

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

ps：当然也可以使用下面的超级管理员权限进行图形化界面修改（当windows使用了）

sudo nautilus

大功告成！！！

多个版本的cuda

如果你需要安装多个版本的cuda（添加第二个cuda的环境变量）
安装好第一个cuda之后，继续安装下一个cuda，之后继续添加环境变量，并建立cuda软链接

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export PATH=$PATH:/usr/local/cuda-11.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1

更换使用的cuda只需要执行下列代码即可

sudo rm -rf /usr/local/cuda                          #删除之前创建的软链接
sudo ln -s /usr/local/cuda-11.1  /usr/local/cuda