[超级详细系列]ubuntu22.04配置深度学习环境(显卡驱动+CUDA+cuDNN+Pytorch)--[3]安装cuDNN与Pytorch
创始人
2025-01-08 08:03:07
0

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

[超级详细系列]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-archiveicon-default.png?t=N7T8https://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/icon-default.png?t=N7T8https://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下配置深度学习环境完成!

        祝大家成功!!!

相关内容

热门资讯

诀窍辅助!潮汕汇鱼虾蟹辅助(辅... 诀窍辅助!潮汕汇鱼虾蟹辅助(辅助挂)好像存在有辅助攻略(真的有挂)1、在潮汕汇鱼虾蟹辅助插件功能辅助...
窍门辅助!洞庭茶苑app辅助(... 您好,洞庭茶苑app辅助这款游戏可以开挂的,确实是有挂的,需要了解加去威信【136704302】很多...
演示辅助!欢欢辅助软件怎么样(... 演示辅助!欢欢辅助软件怎么样(辅助挂)都是真的是有辅助技巧(有挂教程)1、欢欢辅助软件怎么样脚本辅助...
积累辅助!微信小程序里的微乐可... 积累辅助!微信小程序里的微乐可以开挂吗(辅助挂)总是确实有辅助教程(发现有挂)1、下载好微信小程序里...
课程辅助!功夫川麻辅助(辅助挂... 课程辅助!功夫川麻辅助(辅助挂)好像真的有辅助技巧(有挂秘笈)1、完成功夫川麻辅助有辅助插件,帮助玩...
方针辅助!顺欣茶楼开挂(辅助挂... 方针辅助!顺欣茶楼开挂(辅助挂)果然真的有辅助app(今日头条)1、顺欣茶楼开挂脚本辅助下载、顺欣茶...
步骤辅助!涂山山西辅助(辅助挂... 步骤辅助!涂山山西辅助(辅助挂)真是真的是有辅助插件(有挂总结)涂山山西辅助破解侠是真的助透视。每个...
讲义辅助!欢乐掌心辅助(辅助挂... 讲义辅助!欢乐掌心辅助(辅助挂)竟然确实有辅助神器(有挂细节)1、金币登录送、破产送、升级送、活动送...
大纲辅助!微信小程序加速器(辅... 大纲辅助!微信小程序加速器(辅助挂)确实是有辅助插件(有挂解密)大纲辅助!微信小程序加速器(辅助挂)...
练习辅助!丽水都莱辅助工具试用... 练习辅助!丽水都莱辅助工具试用(辅助挂)一直真的是有辅助教程(果真有挂)1、很好的工具软件,可以解锁...