[超级详细系列]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下配置深度学习环境完成! 祝大家成功!!!

相关推荐
stay night481 小时前
DAY40 训练和测试
人工智能·深度学习
SSH_55232 小时前
【大模型】情绪对话模型项目研发
人工智能·python·语言模型
love530love2 小时前
【笔记】在 MSYS2(MINGW64)中安装 python-maturin 的记录
运维·开发语言·人工智能·windows·笔记·python
清醒的兰3 小时前
OpenCV 图像像素的算术操作
人工智能·opencv·计算机视觉
拾忆-eleven3 小时前
NLP学习路线图(十四):词袋模型(Bag of Words)
人工智能·学习·自然语言处理·nlp
sbc-study4 小时前
精英-探索双群协同优化(Elite-Exploration Dual Swarm Cooperative Optimization, EEDSCO)
人工智能
白熊1884 小时前
【机器学习基础】机器学习入门核心算法:XGBoost 和 LightGBM
人工智能·算法·机器学习
微学AI5 小时前
智能穿戴新标杆:SD NAND (贴片式SD卡)与 SOC 如何定义 AI 眼镜未来技术路径
人工智能·ai·sd
拾忆-eleven6 小时前
NLP学习路线图(十五):TF-IDF(词频-逆文档频率)
人工智能·学习·自然语言处理·nlp
G皮T6 小时前
【Python Cookbook】文件与 IO(二)
python·i/o·io·文件·gzip·stringio·bytesio