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

真·手把手教你配置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 同意安装
【ubuntu|真·手把手教你配置ubuntu20.04中cuda+cudnn】重要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/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo 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只需要执行下列代码即可