orin刷机安装torch与torchvision全过程_2.1.0a0+41361538.nv23.06

前言与参考

orin的刷机过程与xavier类似，网上已经有很多教程，但是还是有很多细节的问题多数教程没有交代清楚。详细的步骤可以参考一下博客中的内容，本文只对一些踩坑细节做一些提示。本人的设备是64G的开发者套件版本。

刷机过程：https://blog.csdn.net/weixin_61975705/article/details/127752457

torch与torchvision安装：https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

torch安装nvidia官方教程：https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html

刷机过程

参考博客博主已经对刷机过程进行了详细的展示，需要注意的只有一点：一定要刷最新的版本，不然可能会在安装时出现断链等问题。

torch安装过程

根据JetPack版本不同，需要下载不同的whl文件本地安装，如果在第二篇参考中没有找到自己的版本，可以来到官方教程中，官方教程的方法可能因为网络问题导致下载失败，可以直接进入链接https://developer.download.nvidia.cn/compute/redist/jp/v511，选择pytorch，下载对应的whl文件，其中v511代表JetPack版本，可以通过cat /etc/nv_tegra_release查看自己的JetPack版本。下载后使用以下命令安装：

pip3 install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
1

torchvision安装过程

torchvision需要根据torch的版本进行安装，可以通过以下命令查看torch版本：

python3 -c "import torch; print(torch.__version__)"
1

其实根据上面的文件名也可以发现，本人安装的torch版本是2.1.0。
之后去github 或者gitcode上下载对应版本的torchvision源码，然后进入torchvision目录，使用以下命令安装：
首先将下载下来的压缩包解压，将解压后的文件夹命名为torchvision，然后进入torchvision目录，使用以下命令安装：

export BUILD_VERSION=0.x.0  # 0.x.0代表torchvision的版本号
$ python3 setup.py install --user
1
2

需要等待的时间比较长，耐心等待即可。