From 775399c9a7387fd515bebc223203360d650b793c Mon Sep 17 00:00:00 2001 From: AceofStades Date: Sat, 23 May 2026 04:18:12 +0530 Subject: [PATCH] docs: fix JetPack installation instructions and OOM issues - Add missing \`git clone\` and \`cd\` steps before building the wheel. - Add a note about potential OOM issues on Jetson devices when using Bazel and recommend \`.bazelrc\` limits. - Fix broken code block formatting in the wheel installation step and use a wildcard for the wheel filename." --- docsrc/getting_started/jetpack.rst | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) diff --git a/docsrc/getting_started/jetpack.rst b/docsrc/getting_started/jetpack.rst index 874b526a2e..4ff4b22b8d 100644 --- a/docsrc/getting_started/jetpack.rst +++ b/docsrc/getting_started/jetpack.rst @@ -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