Linux 35.6 + JetPack v5.1.4之 pytorch升级

1. 源由

鉴于之前NVIDIA提供基于JetPack v5.1.4二进制pytorch版本最高为 v2.1.0。

经过不断地各方努力，《Linux 35.6 + JetPack v5.1.4之编译 pytorch》，我们已经有了python3.8.0的 pytorch v2.4.1.

2. 升级

步骤1：获取二进制版本

$ wget https://github.com/SnapDragonfly/pytorch/releases/download/v2.3.1%2Bl4t35.6-cp38-cp38-aarch64/torch-2.3.1+l4t35.6-cp38-cp38-linux_aarch64.whl

步骤2：安装二进制版本

$ sudo pip3 install torch-2.3.1+l4t35.6-cp38-cp38-linux_aarch64.whl

步骤3：获取torchvision

$ git clone https://github.com/SnapDragonfly/vision.git torchvision
$ cd torchvision
$ git checkout nvidia_v0.19.1

关于pytorch和torchvision之间的版本匹配问题，详见：

• Installing previous versions of PyTorch
• torchvision Project
• torchvision Github

步骤4：安装torchvision

$ export BUILD_VERSION=0.19.1
$ sudo python3 setup.py install --user
$ cd ..

步骤5：检查安装版本

$ git clone https://github.com/SnapDragonfly/jetson-fpv.git
$ cd jetson-fpv
$ sudo ./wrapper.sh version
[sudo] password for daniel:
Skipping CMD_KEYMONITOR execution for module 'version'.
Executing command on module version:

Software part of jetson-stats 4.2.12 - (c) 2024, Raffaello Bonghi
Model: NV