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

        祝大家成功!!!

相关内容

热门资讯

微扑克总结!德扑之星能开挂透视... 微扑克总结!德扑之星能开挂透视吗(辅助),总是真的有辅助工具是一款可以让一直输的玩家,快速成为一个“...
WePoKer必备!德扑之星是... 您好,德扑之星是不是有人用挂这款游戏可以开挂的,确实是有挂的,需要了解加微【136704302】很多...
aapoker必备!wepok... 一款 AI 游戏攻略黑科技工具上线,查攻略效率直接开挂。它整合版本更新、装备搭配、BOSS 打法、牌...
菠萝智星德州普及!WePoKe... 菠萝智星德州普及!WePoKer辅助挂怎么下载(透视挂),好像有透视挂脚本是一款可以让一直输的玩家,...
WePoKer揭露!wepok... 您好,wepoker辅助工具好用吗这款游戏可以开挂的,确实是有挂的,需要了解加微【485275054...
德扑之星总结!wepoker有... 很多人使用云端 AI 时,都担心wepoker有挂吗作弊、内部资料上传带来隐私风险。这款本地运行的 ...
红龙PoKer解谜!安卓wep... 您好:安卓wepoker德州透视挂下载这款游戏可以开挂的,确实是有挂的,很多玩家在这款游戏中打牌都会...
wpk分享!wepokerpu... wpk分享!wepokerpus透视挂是真的假的(辅助器),素来真的有辅助器神器是一款可以让一直输的...
aapoker总结!wepok... AI 黑科技赛道又上新,这款智能处理工具堪称效率开挂,专门面向短视频创作者打造。它打通脚本撰写、AI...
微扑克曝光!runit德州透视... 微扑克曝光!runit德州透视软件辅助挂(开挂),一向有开挂插件是一款可以让一直输的玩家,快速成为一...