注意:pytorch和torchvision都不能直接使用pip install进行安装,这样安装的版本是cpu版本,需要去参考链接【3】处下载torch安装包,下载完成后离线安装

pip install path/to/your/whl_file.whl

Jetpack版本与pytorch的部分对应表,如下图,完整对应请查看参考链接【3】:

版本对应表

torch

torchvision

1.13.1

0.14.1

1.13.0

0.14.0

1.12.0

0.13.0

1.11.0

0.12.0

1.10.0

0.11.1

1.9.0

0.10.0

1.8.0

0.9.0

此处有torch1.8-1.10的whl文件以及对应的torchvision文件:

链接:https://pan.baidu.com/s/1YD1y_-v6YppjaULUhZ9rPQ?pwd=zp1h 
提取码:zp1h 
点此下载

安装时会自动安装numpy,特别注意,torch-1.10对应的numpy版本一定要1.19.3的,默认安装的1.19.5的会报错

这些pytorch包仅支持python3.6,使用conda直接创建python3.6时会报错,修改命令为

conda create -n <env_name> python=3.6 -c conda-forge

来正确创建python3.6环境

 参考链接: 

【1】jetson Xavier nx安装torch和torchvision,并解决解决版本不匹配(报错RuntimeError: Couldn‘t load custom C++ ops)的问题_jetson torchvision_hhfhng的博客-CSDN博客

【2】NVIDIA JETSON XAVIER NX (四)安装Pytorch和torchvision_jetson安装pytorch-CSDN博客

【3】PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐