jetson orin 上安装pytorch避坑
jetson orin 上安装pytorch避坑

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版本对应关系和安装方法,真的感动死