Tensorrt on Jetson with python 3.9

Hi,

We have confirmed that bindings for python3.9 can be compiled and installed on JetPack4.6.
Please check the following for the detailed instructions:

1. Building python3.9

$ sudo apt install zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev libsqlite3-dev libbz2-dev
$ wget https://www.python.org/ftp/python/3.9.1/Python-3.9.1.tar.xz
$ tar xvf Python-3.9.1.tar.xz Python-3.9.1/
$ mkdir build-python-3.9.1
$ cd build-python-3.9.1/
$ ../Python-3.9.1/configure --enable-optimizations
$ make -j $(nproc)
$ sudo -H make altinstall
$ cd ../

2. Build cmake 3.13.5

$ sudo apt-get install -y protobuf-compiler libprotobuf-dev openssl libssl-dev libcurl4-openssl-dev
$ wget https://github.com/Kitware/CMake/releases/download/v3.13.5/cmake-3.13.5.tar.gz
$ tar xvf cmake-3.13.5.tar.gz
$ rm cmake-3.13.5.tar.gz
$ cd cmake-3.13.5/
$ ./bootstrap --system-curl
$ make -j$(nproc)
$ echo 'export PATH='${PWD}'/bin/:$PATH' >> ~/.bashrc
$ source ~/.bashrc
$ cd ../

3. Prepare header

$ mkdir python3.9
$ mkdir python3.9/include
$ wget http://ftp.us.debian.org/debian/pool/main/p/python3.9/libpython3.9-dev_3.9.9-2_arm64.deb
$ ar x libpython3.9-dev_3.9.9-2_arm64.deb
$ tar -xvf data.tar.xz
$ cp ./usr/include/aarch64-linux-gnu/python3.9/pyconfig.h python3.9/include/
$ cp -r Python-3.9.1/Include/* python3.9/include/

4. Build TensorRT pybinding

$ git clone https://github.com/pybind/pybind11.git
$ git clone -b release/8.0 https://github.com/NVIDIA/TensorRT.git
$ cd TensorRT
$ git submodule update --init --recursive

Please replay {TensorRT}/python/build.sh with this file: build.sh (2.5 KB)

$ cd python/
$ TRT_OSSPATH=${PWD}/.. EXT_PATH=${PWD}/../.. ./build.sh
$ python3.9 -m pip install build/dist/tensorrt-8.0.1.6-cp39-none-linux_aarch64.whl

Reference:
https://arcanesciencelab.wordpress.com/2021/02/14/building-python-3-9-1-on-jetpack-4-5-and-the-jetson-xavier-nx/
https://github.com/NVIDIA/TensorRT/tree/main/python

Thanks.

4 Likes