-
Flash the Jetson TX2 using JetPack 3.2. Be sure to install
- CUDA 9.0
- OpenCV4Tegra
- cuDNN
- TensorRT 3.0
-
Install TensorFlow on Jetson TX2.
-
Download TensorFlow 1.5.0 / CUDA 9 / cuDNN 7 pip wheel from here.
-
Install TensorFlow using pip
sudo pip install tensorflow-1.5.0rc0-cp27-cp27mu-linux_aarch64.whl
-
-
Install uff converter on Jetson TX2.
-
Download TensorRT 3.0.4 for Ubuntu 16.04 and CUDA 9.0 tar package from https://developer.nvidia.com/nvidia-tensorrt-download.
-
Extract archive
tar -xzf TensorRT-3.0.4.Ubuntu-16.04.3.x86_64.cuda-9.0.cudnn7.0.tar.gz
-
Install uff python package using pip
sudo pip install TensorRT-3.0.4/uff/uff-0.2.0-py2.py3-none-any.whl
-
-
Clone and build this project
git clone --recursive https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification.git cd tf_to_trt_image_classification mkdir build cd build cmake .. make cd ..