[超等具体系列]ubuntu22.04设置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[3]安装cuDNN与Pytorch

[复制链接]
发表于 2026-4-24 08:56:54 | 显示全部楼层 |阅读模式
本次设置过程的三篇博文分享分别为为:
[超等具体系列]ubuntu22.04设置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[1]安装显卡驱动
[超等具体系列]ubuntu22.04设置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[2]安装Anaconda与CUDA
[超等具体系列]ubuntu22.04设置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[3]安装cuDNN与Pytorch(本篇)

1、设置cuDNN

点击下方链接进入cuDNN官网:
https://developer.nvidia.com/rdp/cudnn-archivehttps://developer.nvidia.com/rdp/cudnn-archive
https://developer.nvidia.com/rdp/cudnn-archive
        根据之前的CUDA版本选择对应的cuDNN版本,具体对应规则可以参考Franpper写的另一篇博客:
深度学习环境设置:CUDA、cuDNN 和 PyTorch 版本的选择与搭配指南(发起收藏)
        由于Franpper之前安装了12.0版本的CUDA,以是这里选择cuDNN v8.9.5举行安装。


        下载完成后,打开下载路径,右键选择在终端打开

        通过下面指令解压文件
  1. tar -xvf cudnn-linux-x86_64-8.9.5.30_cuda12-archive.tar.xz
复制代码
        然后,利用下面两条指令复制cuDNN文件夹下的文件到/usr/local/cuda-11.0/lib64/和/usr/local/cuda-11.0/include/中(以本身现实安装路径为准)。
  1. sudo cp cudnn-linux-x86_64-8.9.5.30_cuda12-archive/include/cudnn*.h /usr/local/cuda/include
  2. sudo cp -p cudnn-linux-x86_64-8.9.5.30_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64
  3. sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
复制代码

        到此为止cuDNN设置完毕。
2、安装PyTorch

        点击下方链接进入PyTorch官网,选择对应的torch版本。对应规则也可以参考上面的版本选择与搭配指南。
Previous PyTorch Versions | PyTorchInstalling previous versions of PyTorchhttps://pytorch.org/get-started/previous-versions/
https://pytorch.org/get-started/previous-versions/         进入本身创建好的环境中
  1. conda activate franpper
复制代码

        输入复制而来的指令 
  1. pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
复制代码
         期待一段时间后,安装完毕

        通过下面的指令举行测试是否安装乐成
  1. import torch
  2. torch.cuda.is_available()
  3. print(torch.__version__)
  4. print(torch.version.cuda)
  5. print(torch.backends.cudnn.version())
复制代码

        大功告成!!!
        至此,Ubuntu下设置深度学习环境完成!
        祝各人乐成!!!

本帖子中包含更多资源

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

×
回复

使用道具 举报

登录后关闭弹窗

登录参与点评抽奖  加入IT实名职场社区
去登录
快速回复 返回顶部 返回列表