Needs update for Jetson Xavier

NVIDIA-AI-IOT / tf_to_trt_image_classification

Image classification with NVIDIA TensorRT from TensorFlow models.

BSD 3-Clause "New" or "Revised" License

454 stars 155 forks source link

Needs update for Jetson Xavier #28

Open NathanMOlson opened 5 years ago

NathanMOlson commented 5 years ago

Jetson Xavier uses TensorRT 5 and CUDA 10, which aren't supported by the current setup instructions.

ghost commented 5 years ago

Thanks for bringing this up.

We have been able to run this repository on Xavier with a few modifications.

The branch trt_4plus has a small patch that addresses compatibility with TensorRT 4+.

Please let me know if you check out this branch and it works for you. We will work on updating the setup instructions accordingly.

NathanMOlson commented 5 years ago

Great, thanks. The trt_4plus patch worked for me, but based on profiling it seems the NVDLA is not being used. Does this branch compile for NVDLA? Is there a way to see NVDLA usage?