首页
找靠谱产品
找解决方案
找靠谱公司
找案例
找对的人
专家智库
悬赏任务
SAAS
ToB门户
了解全球最新的ToB事件
论坛
潜水/灌水快乐,沉淀知识,认识更多同行。
ToB圈子
加入IT圈,遇到更多同好之人。
微博
Follow
记录
Doing
博客
Blog
文库
业界最专业的IT文库,上传资料也可以赚钱
下载
分享
Share
排行榜
Ranklist
相册
Album
应用中心
qidao123.com ToB IT社区-企服评测·应用市场
»
论坛
›
软件与程序人生
›
DevOps与敏捷开发
›
[超等具体系列]ubuntu22.04设置深度学习环境(显卡驱动+C ...
返回列表
发新帖
[超等具体系列]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举行安装。
下载完成后,打开下载路径,右键选择在
终端
打开
通过下面指令解压文件
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/中(以本身现实安装路径为准)。
sudo cp cudnn-linux-x86_64-8.9.5.30_cuda12-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -p cudnn-linux-x86_64-8.9.5.30_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64
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/ 进入本身创建好的环境中
conda activate franpper
复制
代码
输入复制而来的指令
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
复制代码
期待一段时间后,安装完毕
通过下面的指令举行测试是否安装乐成
import torch
torch.cuda.is_available()
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
复制代码
大功告成!!!
至此,Ubuntu下设置深度学习环境完成!
祝各人乐成!!!
本帖子中包含更多资源
您需要
登录
才可以下载或查看,没有账号?
立即注册
×
回复
使用道具
举报
返回列表
傲渊山岳
+ 我要发帖
登录后关闭弹窗
登录参与点评抽奖 加入IT实名职场社区
去登录
微信订阅号
微信服务号
微信客服(加群)
H5
小程序
快速回复
返回顶部
返回列表