@@ -60,8 +60,22 @@ System Preparation
60
60
sudo cp -a libcusparse_lt-linux-sbsa-0.5.2.1-archive/include/* /usr/local/cuda/include/
61
61
sudo cp -a libcusparse_lt-linux-sbsa-0.5.2.1-archive/lib/* /usr/local/cuda/lib64/
62
62
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.
65
79
66
80
Build Environment Setup
67
81
=======================
@@ -92,25 +106,22 @@ Build Environment Setup
92
106
# Can only install the torch and torchvision wheel from the JPL repo which is built specifically for JetPack 6.2
93
107
python -m pip install torch==2.8.0 torchvision==0.23.0 --index-url=https://pypi.jetson-ai-lab.io/jp6/cu126
94
108
109
+ 4. **Build the Wheel **:
95
110
96
- Building the Wheel
97
- ==================
111
+ .. code-block :: sh
98
112
99
- .. code-block :: sh
100
- python setup.py bdist_wheel --jetpack
113
+ python setup.py bdist_wheel --jetpack
101
114
102
- Installation
103
- ============
115
+ 5. **Install the Wheel **:
104
116
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
107
120
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
109
122
110
- Post-Installation Verification
111
- ==============================
123
+ 6. **Verify installation by importing in Python **:
112
124
113
- Verify installation by importing in Python:
114
125
.. code-block :: python
115
126
116
127
# verify whether the torch-tensorrt can be imported
0 commit comments