Skip to content

[Docs] Fix JetPack installation gaps, formatting, and OOM issues #4292

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
AceofStades wants to merge 1 commit into
base: main
Choose a base branch
from
+16 −6
Open

[Docs] Fix JetPack installation gaps, formatting, and OOM issues #4292

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 16 additions & 6 deletions docsrc/getting_started/jetpack.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -106,21 +106,31 @@ Build Environment Setup
# Can only install the torch and torchvision wheel from the JPL repo which is built specifically for JetPack 6.2
python -m pip install torch==2.8.0 torchvision==0.23.0 --index-url=https://pypi.jetson-ai-lab.io/jp6/cu126

4. **Build the Wheel**:
4. **Clone the Repository**:

.. code-block:: sh

git clone https://github.com/pytorch/TensorRT.git
cd TensorRT

5. **Build the Wheel**:

.. note::
Compiling with Bazel on shared-memory edge devices like the Jetson Nano or Orin can cause out-of-memory (OOM) crashes. To avoid this, limit Bazel's CPU and memory usage by adding limits in your ``.bazelrc`` (e.g., adding ``build --jobs=2``) or by creating a temporary swapfile.

.. code-block:: sh

python setup.py bdist_wheel --jetpack

5. **Install the Wheel**:
6. **Install the Wheel**:

.. code-block:: sh

# you will be able to find the wheel in the dist directory
cd dist
python -m pip install torch_tensorrt-2.8.0.dev0+d8318d8fc-cp310-cp310-linux_aarch64.whl
# you will be able to find the wheel in the dist directory
cd dist
python -m pip install torch_tensorrt-*.whl

6. **Verify installation by importing in Python**:
7. **Verify installation by importing in Python**:

.. code-block:: python

Expand Down
Loading