message: Installs TensorRT
diff: @@ -7,6 +7,12 @@ LABEL maintainer="Jan Prach <jendap@google.com>"
  RUN cp -P /usr/include/cudnn.h /usr/local/cuda/include
  RUN cp -P /usr/lib/x86_64-linux-gnu/libcudnn* /usr/local/cuda/lib64
  +# Installs TensorRT, which is not included in NVIDIA Docker containers.
  +RUN apt-get update \
  + && apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda10.0 \
  + && apt-get update \
  + && apt-get install -y --no-install-recommends libnvinfer-dev=5.0.2-1+cuda10.0
  +
  # Copy and run the install scripts.
  COPY install/*.sh /install/
  ARG DEBIAN_FRONTEND=noninteractive
  