RomainGsd/TensorFlow-CTR-arm64

Tensorflow container with Tensorrt support for arm64 - CUDA(11.0) CUDNN(8.1.1) TensorRT(7.1)

TensorFlow for arm64

Tensorflow container with Tensorrt support for arm64 - CUDA(11.1) CUDNN(8.1.1) TensorRT(7.2.1)

Follow this guide to install tensorRT :

https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html

TensorRT DEB file can be downloaded from here:

https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/7.2.1/local_repos/nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.1.6-ga-20201007_1-1_arm64.deb