Ubuntu20.04 安装 cudatookit 12.2 + cudnn 安装

金歌  金牌会员 | 2024-9-20 08:03:08 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 919|帖子 919|积分 2757

最简约的部署Ubuntu20.04深度学习环境的教程

1. 安装Ubuntu20.04 系统

B站详细的安装教程
简约安装版
2. 安装Nvidia显卡驱动

我参考了各种资料,重装系统,完善办理开机显示器黑屏无法进入桌面的环境
黑屏题目主要是由linux内核更新导致,nvidia 不兼容 ubuntu的图像显示,以是很容易出现黑屏环境
在安装nvidia驱动前一定要使用命令:
  1. # 一定要保持当前的内核,也就是安装 NVIDIA 驱动时用的内核
  2. sudo apt-mark hold linux-image-generic linux-headers-generic
复制代码
即保持当前系统的内核不更新之后,再去安装 NVIDIA 驱动,成功办理题目
打开应用程序里面的 软件与更新 选择最适合自己的版本,然后直接点击 应用更改,等待一会

安装完毕后,直接重新启动,然后输入nvidia-smi 成功出现以下画面:

3. 安装 cudatookit 12.2

由于显卡最高支持到CUDA 12.2
这个版本从前的CUDA都是兼容的,怕以后再升级贫苦,以是就直接安装支持的最高版本了
CUDA Toolkit Archive的网址为 https://developer.nvidia.com/cuda-toolkit-archive
点击CUDA Toolkit 12.2.0下载相应版本的CUDA12.2.0(选自己想要安装的版本即可,这里以12.2.0为例)


为了少输入点指令,以是就选择了runfile,感觉是最简单的方法了
运行上面指令后,会弹出如下界面,点击Continue,然后再输入accept

在弹出的界面中通过Enter键,取消Driver和535 的安装,然后点击Install,等待
设置CUDA的环境变量
CUDA安装完成后,需要设置变量环境才气正常使用。起首在终端输入
  1. sudo gedit ~/.bashrc
复制代码
打开.bashrc文件。然后,在.bashrc文件的末了添加以下CUDA环境变量设置信息
  1. export PATH=$PATH:/usr/local/cuda/bin  
  2. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
  3. export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
复制代码
末了 ctrl+s 保存
革新设置
  1. source ~/.bashrc
复制代码
检察并且测试安装
  1. nvcc -V
复制代码

检察到CUDA的安装信息,如下图所示,至此CUDA安装成功!
4. 安装cuDNN

cuDNN的网址为 https://developer.nvidia.com/rdp/cudnn-download
选择Ubuntu20.04系统下,CUDA12.X 对应的cuDNN v8.9.6版本,如下图所示:

下载对应版本的 deb 文件,在下载的文件夹里面,使用命令安装
  1. sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.6.50_1.0-1_amd64.deb
复制代码
安装完成会显示,反面的操作命令,执行对应的命令就行:
  1. sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.6.50/cudnn-local-1998375D-keyring.gpg /usr/share/keyrings/
复制代码
若没有显示提示命令则执行通用命令:
  1. sudo cp /var/cudnn-local-*/cudnn-*-keyring.gpg /usr/share/keyrings/
复制代码
安装对应的依靠包
进入文件夹cd /var/cudnn-local-repo-ubuntu2204-8.9.6.50/,安装对应的依靠包,这些依靠包直接下载安装会失败,都包罗在deb文件中,进入上面提到的文件夹安装即可:
  1. sudo dpkg -i libcudnn8_8.9.6.50-1+cuda12.2_amd64.deb
  2. sudo dpkg -i libcudnn8-dev_8.9.6.50-1+cuda12.2_amd64.deb
  3. sudo dpkg -i libcudnn8-samples_8.9.6.50-1+cuda12.2_amd64.deb
复制代码
验证安装成功
  1. cd /usr/src/cudnn_samples_v8        #  进入文件夹cd /usr/src/cudnn_samples_v8
  2. cp -r /usr/src/cudnn_samples_v8/ $HOME        # 将 cuDNN 示例复制到用户文件
  3. cd $HOME/cudnn_samples_v8/mnistCUDNN
  4. make clean && make        # 编译 cuDNN mnisiCUDNN 示例
  5. ./mnistCUDNN        # 运行 mnistCUDNN 示例
复制代码

返回 Test passed! 则证明安装成功!!!开始深度学习炼丹之旅吧!!!
参考文献

ubuntu20.04安装NVIDIA显卡以及重启黑屏的一种办理方案
Ubuntu 20.04-NVIDIA显卡驱动-安装和卸载-办理黑屏题目
Ubuntu系统安装nvidia(英伟达)驱动后,开机显示器黑屏无法进入桌面
Ubuntu20.04下CUDA、cuDNN的详细安装与设置过程(图文)
Ubuntu20.04设置NVIDIA+CUDA12.2+CUDNN【附全部下载资源】【亲测有效】【非常详细】
Ubuntu 22.04 安装 CUDA 和 cuDNN 及验证

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

金歌

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表