Skip to content

Commit 863eece

Browse files
committed
update docs
1 parent 545364e commit 863eece

File tree

1 file changed

+25
-14
lines changed

1 file changed

+25
-14
lines changed

docsrc/getting_started/jetpack.rst

Lines changed: 25 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,22 @@ System Preparation
6060
sudo cp -a libcusparse_lt-linux-sbsa-0.5.2.1-archive/include/* /usr/local/cuda/include/
6161
sudo cp -a libcusparse_lt-linux-sbsa-0.5.2.1-archive/lib/* /usr/local/cuda/lib64/
6262
63-
Building Torch-TensorRT
64-
***********************
63+
Installation Torch-TensorRT in JetPack
64+
*************************************
65+
66+
You can directly install the torch-tensorrt wheel from the JPL repo which is built specifically for JetPack 6.2.
67+
68+
.. code-block:: sh
69+
# verify tensorrt 10.3 is already installed via jetpack installation process
70+
python -m pip list | grep tensorrt
71+
# install torch-tensorrt wheel from JPL repo which is built specifically for JetPack 6.2
72+
python -m pip install torch==2.8.0 torch_tensorrt==2.8.0 torchvision==0.24.0 --extra-index-url https://pypi.jetson-ai-lab.io/jp6/cu126
73+
74+
75+
Building Torch-TensorRT in JetPack
76+
*********************************
77+
78+
You can also build the torch-tensorrt wheel from the source code on your own.
6579

6680
Build Environment Setup
6781
=======================
@@ -92,25 +106,22 @@ Build Environment Setup
92106
# Can only install the torch and torchvision wheel from the JPL repo which is built specifically for JetPack 6.2
93107
python -m pip install torch==2.8.0 torchvision==0.23.0 --index-url=https://pypi.jetson-ai-lab.io/jp6/cu126
94108
109+
4. **Build the Wheel**:
95110

96-
Building the Wheel
97-
==================
111+
.. code-block:: sh
98112
99-
.. code-block:: sh
100-
python setup.py bdist_wheel --jetpack
113+
python setup.py bdist_wheel --jetpack
101114
102-
Installation
103-
============
115+
5. **Install the Wheel**:
104116

105-
.. code-block:: sh
106-
# you will be able to find the wheel in the dist directory, has platform name linux_tegra_aarch64
117+
.. code-block:: sh
118+
119+
# you will be able to find the wheel in the dist directory
107120
cd dist
108-
python -m pip install torch_tensorrt-2.8.0.dev0+d8318d8fc-cp310-cp310-linux_tegra_aarch64.whl
121+
python -m pip install torch_tensorrt-2.8.0.dev0+d8318d8fc-cp310-cp310-linux_aarch64.whl
109122

110-
Post-Installation Verification
111-
==============================
123+
6. **Verify installation by importing in Python**:
112124

113-
Verify installation by importing in Python:
114125
.. code-block:: python
115126
116127
# verify whether the torch-tensorrt can be imported

0 commit comments

Comments
 (0)