Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]

CUDA安装

下载cuda

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

图片[1]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

如果无法查看,则说明尚未安装nvidia驱动,点击附加驱动,选择对应版本的驱动即可自动下载。

图片[2]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda,选择20.04的版本即可。

图片[3]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

gcc降级(cuda高版本不需要)

Ubuntu20.04自带的gcc版本为9.7.0,需要添加gcc7才可安装cuda10.2,输入命令安装gcc7

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

查看gcc版本,可以看到目前系统中存在7和9两个版本

图片[4]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

使用update-alternatives进行版本切换,输入以下命令:

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命令查看gcc的默认版本,可以看到当前默认gcc版本为7,即切换成功。

图片[5]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

安装cuda

sudo sh cuda_10.2.89_440.33.01_linux.run

图片[6]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

点击continue

图片[7]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

输入accept

图片[8]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install

配置环境变量

输入gedit ~/.bashrc命令打开文件,在文件结尾输入以下语句,保存。

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

更新环境变量配置

 source ~/.bashrc

至此cuda安装完成,输入nvcc -V命令查看cuda信息

图片[9]-Ubuntu 20.04 CUDA&cuDNN安装方法[通俗易懂]-九桑

安装cuDNN

https://developer.nvidia.com/rdp/cudnn-archive下载相应版本的cudnn,需要登陆。

图片

下载解压之后,将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹,将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:

cd cudnn-linux-x86_64-8.9.7.29_cuda12-archive  # 更换成你下载的版本
sudo cp include/* /usr/local/cuda-11.7/include
sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64
# 依次输入这两行代码到终端即可。这里的cuda-版本和上面一样按自己的来。
# cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar.xz 文件名也是按自己的来。

© 版权声明
THE END
喜欢就支持一下吧
点赞13 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容