HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update Jetson Doc with torch and torchvision for JetPack 5.1.2 (#18444)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Lakshantha Dissanayake 2024-12-30 06:38:46 -08:00 • committed by GitHub
parent 842f88dc84
commit 8aec122bb9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

19
docs/en/guides/nvidia-jetson.md
View file

@ -177,7 +177,7 @@ pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/onnxr
`pip install numpy==1.23.5` `pip install numpy==1.23.5`
### Run on JetPack 5.x ### Run on JetPack 5.1.2
#### Install Ultralytics Package #### Install Ultralytics Package
@ -213,22 +213,11 @@ The above ultralytics installation will install Torch and Torchvision. However,
pip uninstall torch torchvision pip uninstall torch torchvision
``` ```
2. Install PyTorch 2.1.0 according to JP5.1.3 2. Install `torch 2.1.0` and `torchvision 0.16.2` according to JP5.1.2
```bash ```bash
sudo apt-get install -y libopenblas-base libopenmpi-dev pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
wget https://developer.download.nvidia.com/compute/redist/jp/v512/pytorch/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl -O torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.16.2+c6f3977-cp38-cp38-linux_aarch64.whl
pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
```
3. Install Torchvision v0.16.2 according to PyTorch v2.1.0
```bash
sudo apt install -y libjpeg-dev zlib1g-dev
git clone https://github.com/pytorch/vision torchvision
cd torchvision
git checkout v0.16.2
python3 setup.py install --user
``` ```
!!! note !!! note