[超级详细系列]ubuntu22.04配置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[3]安装cuDNN与Pytorch

本次配置过程的三篇博文分享分别为为:

[超级详细系列]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![icon-default.png?t=N7T8](https://csdnimg.cn/release/blog_editor_html/release2.3.6/ckeditor/plugins/CsdnLink/icons/icon-default.png)https://developer.nvidia.com/rdp/cudnn-archive](https://developer.nvidia.com/rdp/cudnn-archive "https://developer.nvidia.com/rdp/cudnn-archivehttps://developer.nvidia.com/rdp/cudnn-archive") 根据之前的CUDA版本选择对应的cuDNN版本,具体对应规则可以参考Franpper写的另一篇博客: [深度学习环境配置:CUDA、cuDNN 和 PyTorch 版本的选择与搭配指南(建议收藏)](https://blog.csdn.net/weixin_58283091/article/details/133349022?spm=1001.2014.3001.5502 "深度学习环境配置:CUDA、cuDNN 和 PyTorch 版本的选择与搭配指南(建议收藏)") 由于Franpper之前安装了12.0版本的CUDA,所以这里选择cuDNN v8.9.5进行安装。 ![](https://img-blog.csdnimg.cn/direct/9c9589a2b7bb4ebb9f987f64cabb33a1.png) 下载完成后,打开下载路径,右键选择在终端打开 ![](https://img-blog.csdnimg.cn/direct/ec630c4b51b1495fa04d2ecbe365bbc5.png) 通过下面指令解压文件 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* ![](https://img-blog.csdnimg.cn/direct/fe3eee4c4df74323bae85b112a1d0eec.png) 到此为止cuDNN配置完毕。 ### 2、安装PyTorch 点击下方链接进入PyTorch官网,选择对应的torch版本。对应规则也可以参考上面的版本选择与搭配指南。 [Previous PyTorch Versions \| PyTorchInstalling previous versions of PyTorchhttps://pytorch.org/get-started/previous-versions/![icon-default.png?t=N7T8](https://csdnimg.cn/release/blog_editor_html/release2.3.6/ckeditor/plugins/CsdnLink/icons/icon-default.png)https://pytorch.org/get-started/previous-versions/](https://pytorch.org/get-started/previous-versions/ "Previous PyTorch Versions | PyTorchInstalling previous versions of PyTorchhttps://pytorch.org/get-started/previous-versions/") 进入自己创建好的环境中 conda activate franpper ![](https://img-blog.csdnimg.cn/direct/9999ed88ea2a4ed18b59e8a849c597d8.png) 输入复制而来的指令 pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118 ![](https://img-blog.csdnimg.cn/direct/fa47fb8e26fb469a8fb986c20a5a0fc7.png) 等待一段时候后,安装完毕 ![](https://img-blog.csdnimg.cn/direct/ada020f18878469894cf08d08536ce5a.png) 通过下面的指令进行测试是否安装成功 import torch torch.cuda.is_available() print(torch.__version__) print(torch.version.cuda) print(torch.backends.cudnn.version()) ![](https://img-blog.csdnimg.cn/direct/c3648c98351b49e99d2c2c1f470293e8.png) 大功告成!!! 至此,Ubuntu下配置深度学习环境完成! 祝大家成功!!!

相关推荐
sali-tec38 分钟前
C# 基于halcon的视觉工作流-章56-彩图转云图
人工智能·算法·计算机视觉·c#
梦想画家1 小时前
基于PyTorch的时间序列异常检测管道构建指南
人工智能·pytorch·python
Elastic 中国社区官方博客2 小时前
在 Elasticsearch 中使用 Mistral Chat completions 进行上下文工程
大数据·数据库·人工智能·elasticsearch·搜索引擎·ai·全文检索
一碗绿豆汤2 小时前
机器学习第二阶段
人工智能·机器学习
PythonFun2 小时前
OCR图片识别翻译工具功能及源码
python·ocr·机器翻译
用什么都重名2 小时前
DeepSeek-OCR 深度解析
人工智能·ocr·deepseek-ocr
河南骏3 小时前
RAG_检索进阶
人工智能·深度学习
虫师c3 小时前
Python浪漫弹窗程序:Tkinter实现动态祝福窗口教程
python·tkinter·动画效果·gui编程·弹窗效果
容器( ु⁎ᴗ_ᴗ⁎)ु.。oO3 小时前
Magentic-ui 学习
学习
_李小白3 小时前
【OPENGL ES 3.0 学习笔记】延伸阅读:VAO与VBO
笔记·学习·elasticsearch