test

pytorch · Nov 7, 2024 · 2455882 · 2455882
1 parent 1626f0c
commit 2455882
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/docsrc/getting_started/jetpack.rst b/docsrc/getting_started/jetpack.rst
@@ -84,6 +84,11 @@ Install pip and required python packages:
     wget https://bootstrap.pypa.io/get-pip.py
     python get-pip.py
 
+.. code-block:: sh
+
+   # install pytorch from nvidia jetson distribution: https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch
+   python -m pip install torch https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
+
 .. code-block:: sh
 
     # install required python packages

diff --git a/toolchains/jp_workspaces/requirements.txt b/toolchains/jp_workspaces/requirements.txt
@@ -2,5 +2,3 @@ setuptools==70.2.0
 numpy<2.0.0
 packaging
 pyyaml
-torch
---index-url https://developer.download.nvidia.com/compute/redist/jp/v61/pytorch/torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl