备忘
备忘

jetson orin 上安装pytorch避坑

第一个坑

安装官网教程直接安装 pip3 install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cu118

不管什么版本,都用不了cuda

找到nvidia官网论坛上有专门的介绍jetson板子的驱动:PyTorch for Jetson – Jetson & Embedded Systems / Jetson Nano – NVIDIA Developer Forums

然后下载whl后直接安装,下载文件名比较长

第二个坑 文件名不能修改

修改之后安装的时候会报错,报的就是文件名不对,也不知道这个设计思路到底是为啥

安装完torch之后,可以看到这个版本比较特殊,好歹cuda可以用了

使用yolo的时候下一个坑出现了,ultralytics 需要torchvision模块,想当然的就 pip install torchvision,结果把刚安装的torch给卸载了,重新装上了torch 2.1.0版本,一夜回到解放前

只能找jetson板子对应的torch vision模块,找了半天,竟然在刚才下载torch的页面上,而且还贴心的给出了torch和torchvision版本对应关系和安装方法,真的感动死

安装历史版本CUDA以及cuDNN

英伟达官网默认都是最新的,想下载历史版本一时不好找,记录一下地址

CUDA Toolkit Archive | NVIDIA Developer

以上是cuda文件,如果是安装cuDnn用于神经网络训练,下载地址是

cuDNN Archive | NVIDIA Developer

如果是纯粹的显卡驱动,下载地址是

Official Drivers | NVIDIA

安装cuda方法在下载页面有提示,一步步照着做就行,但是cuDNN的安装方法下载页面并没有提示,文档地址在这里

Installation Guide :: NVIDIA cuDNN Documentation

以上3种关系需要自己搞清楚,否则后果自负

在使用tensorflow过程中,有可能会提示缺少zlibdll,需要单独下载,然后放到cuda安装目录下,

下载地址为:ZLIB DLL Home Page — ZLIB DLL 主页 (winimage.com)

教程参考:【Error】Could not locate zlibwapi.dll. Please make sure it is in your library path!_MangoloD的博客-CSDN博客

_________________________________________________________________

对于torch的安装也是一个头疼的问题,需要与cuda版本对应,如果是gpu版本,需要指定版本号,比如pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html

可以参照一下帖子:GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~_清华源安装torch-CSDN博客