From 8aec122bb951b1e1e3d75c4ef264ad9cf08d6346 Mon Sep 17 00:00:00 2001 From: Lakshantha Dissanayake Date: Mon, 30 Dec 2024 06:38:46 -0800 Subject: [PATCH] Update Jetson Doc with `torch` and `torchvision` for JetPack 5.1.2 (#18444) Co-authored-by: Glenn Jocher --- docs/en/guides/nvidia-jetson.md | 19 ++++--------------- 1 file changed, 4 insertions(+), 15 deletions(-) diff --git a/docs/en/guides/nvidia-jetson.md b/docs/en/guides/nvidia-jetson.md index 1a42b786..c32b77e9 100644 --- a/docs/en/guides/nvidia-jetson.md +++ b/docs/en/guides/nvidia-jetson.md @@ -177,7 +177,7 @@ pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/onnxr `pip install numpy==1.23.5` -### Run on JetPack 5.x +### Run on JetPack 5.1.2 #### Install Ultralytics Package @@ -213,22 +213,11 @@ The above ultralytics installation will install Torch and Torchvision. However, 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 - sudo apt-get install -y libopenblas-base libopenmpi-dev - 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 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 + 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/torchvision-0.16.2+c6f3977-cp38-cp38-linux_aarch64.whl ``` !!! note