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

Update JetPack 5 torch and torchvision packages (#19098)

Browse source
This commit is contained in:
Lakshantha Dissanayake 2025-02-06 19:16:20 -08:00 • committed by GitHub
parent 7aea6383d1
commit c526a652ab
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
docker/Dockerfile-jetson-jetpack5
View file

@ -34,8 +34,8 @@ ADD https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n.pt .
RUN python3 -m pip install --upgrade pip uv RUN python3 -m pip install --upgrade pip uv
RUN uv pip install --system \ RUN uv pip install --system \
https://github.com/ultralytics/assets/releases/download/v0.0.0/onnxruntime_gpu-1.18.0-cp38-cp38-linux_aarch64.whl \ https://github.com/ultralytics/assets/releases/download/v0.0.0/onnxruntime_gpu-1.18.0-cp38-cp38-linux_aarch64.whl \
https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl \ https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.2.0-cp38-cp38-linux_aarch64.whl \
https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.16.2+c6f3977-cp38-cp38-linux_aarch64.whl https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.17.2+c1d70fe-cp38-cp38-linux_aarch64.whl
RUN uv pip install --system -e ".[export]" RUN uv pip install --system -e ".[export]"

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

@ -213,11 +213,11 @@ The above ultralytics installation will install Torch and Torchvision. However,
pip uninstall torch torchvision pip uninstall torch torchvision
``` ```
2. Install `torch 2.1.0` and `torchvision 0.16.2` according to JP5.1.2 2. Install `torch 2.2.0` and `torchvision 0.17.2` according to JP5.1.2
```bash ```bash
pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.2.0-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 https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.17.2+c1d70fe-cp38-cp38-linux_aarch64.whl
``` ```
!!! note !!! note