TRTorch on jetson xavier agx

Robotics & Edge Computing Jetson Systems Jetson Xavier NX
,
5

Hi,

Please try the following to compile Trtorch with JetPack 4.5.

1. Pull docker

$ sudo docker run -it --rm --runtime nvidia --network host -v /home/nvidia/workspace:/home/nvidia/workspace nvcr.io/nvidia/l4t-pytorch:r32.5.0-pth1.7-py3

2. Build bazel

$ apt-get update
$ apt-get install -y pkg-config zip g++ zlib1g-dev unzip openjdk-8-jdk
$ wget https://github.com/bazelbuild/bazel/releases/download/4.0.0/bazel-4.0.0-dist.zip
$ unzip bazel-4.0.0-dist.zip -d bazel-4.0.0-dist
$ cd bazel-4.0.0-dist
$ EXTRA_BAZEL_ARGS="--host_javabase=@local_jdk//:jdk" ./compile.sh
$ cp output/bazel /usr/bin/

3. Clone TRTorch source

$ cd /home/nvidia/workspace
$ git clone https://github.com/NVIDIA/TRTorch.git

4. Apply changes

Update WORKSPACE with WORKSPACE (5.1 KB)
Apply following change

diff --git a/py/setup.py b/py/setup.py
index 1e6ca6e..218c39d 100644
--- a/py/setup.py
+++ b/py/setup.py
@@ -204,7 +204,7 @@ setup(name='trtorch',
       long_description=long_description,
       ext_modules=ext_modules,
       install_requires=[
-          'torch==1.7.1',
+          'torch==1.7.0',
       ],
       setup_requires=[],
       cmdclass={

5. Compile and install

$ cd TRTorch/py
$ python3 setup.py bdist_wheel --use-cxx11-abi
$ pip3 install dist/trtorch-0.2.0a0-cp36-cp36m-linux_aarch64.whl

Thanks.