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

相关推荐
会飞的老朱33 分钟前
医药集团数智化转型,智能综合管理平台激活集团管理新效能
大数据·人工智能·oa协同办公
聆风吟º2 小时前
CANN runtime 实战指南:异构计算场景中运行时组件的部署、调优与扩展技巧
人工智能·神经网络·cann·异构计算
寻星探路2 小时前
【深度长文】万字攻克网络原理:从 HTTP 报文解构到 HTTPS 终极加密逻辑
java·开发语言·网络·python·http·ai·https
Codebee4 小时前
能力中心 (Agent SkillCenter):开启AI技能管理新时代
人工智能
聆风吟º5 小时前
CANN runtime 全链路拆解:AI 异构计算运行时的任务管理与功能适配技术路径
人工智能·深度学习·神经网络·cann
uesowys5 小时前
Apache Spark算法开发指导-One-vs-Rest classifier
人工智能·算法·spark
AI_56785 小时前
AWS EC2新手入门:6步带你从零启动实例
大数据·数据库·人工智能·机器学习·aws
User_芊芊君子5 小时前
CANN大模型推理加速引擎ascend-transformer-boost深度解析:毫秒级响应的Transformer优化方案
人工智能·深度学习·transformer
ValhallaCoder5 小时前
hot100-二叉树I
数据结构·python·算法·二叉树
执笔论英雄5 小时前
【大模型学习cuda】入们第一个例子-向量和
学习