From 025abf996dfce0d9c7e2957e0efcc28a936454f3 Mon Sep 17 00:00:00 2001 From: George Wu Date: Wed, 7 Apr 2021 08:48:43 -0700 Subject: [PATCH] fix for using tensorrt:20.12 base image (#7264) --- dockerfiles/Dockerfile.tensorrt | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/dockerfiles/Dockerfile.tensorrt b/dockerfiles/Dockerfile.tensorrt index 88ddd6369a..87e088e414 100644 --- a/dockerfiles/Dockerfile.tensorrt +++ b/dockerfiles/Dockerfile.tensorrt @@ -5,8 +5,7 @@ # Dockerfile to run ONNXRuntime with TensorRT integration # nVidia TensorRT Base Image -FROM nvcr.io/nvidia/tensorrt:20.09-py3 -MAINTAINER Vinitra Swamy "viswamy@microsoft.com" +FROM nvcr.io/nvidia/tensorrt:20.12-py3 ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime ARG ONNXRUNTIME_BRANCH=master @@ -17,7 +16,6 @@ RUN unattended-upgrade WORKDIR /code ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:/code/cmake-3.14.3-Linux-x86_64/bin:/opt/miniconda/bin:${PATH} -ENV LD_LIBRARY_PATH /opt/miniconda/lib:$LD_LIBRARY_PATH # Prepare onnxruntime repository & build onnxruntime with TensorRT RUN git clone --single-branch --branch ${ONNXRUNTIME_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime &&\