From a9549b57b1e0aba2167207eb01312d6ea8790049 Mon Sep 17 00:00:00 2001 From: Per Goncalves da Silva Date: Wed, 15 May 2019 10:30:34 +0200 Subject: [PATCH] Pins libnvinfer versions --- ci/docker/install/tensorrt.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ci/docker/install/tensorrt.sh b/ci/docker/install/tensorrt.sh index 1950cad0b52f..0616830c6b98 100755 --- a/ci/docker/install/tensorrt.sh +++ b/ci/docker/install/tensorrt.sh @@ -44,5 +44,6 @@ echo "TensorRT build enabled. Installing TensorRT." wget -qO tensorrt.deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda10.0_1-1_amd64.deb dpkg -i tensorrt.deb apt-get update -apt-get install -y --allow-downgrades libnvinfer-dev +apt-get install -y libnvinfer5=5.1.2-1+cuda10.0 +apt-get install -y libnvinfer-dev=5.1.2-1+cuda10.0 rm tensorrt.deb