Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Pins libnvinfer versions
Browse files Browse the repository at this point in the history
  • Loading branch information
@perdasilva
perdasilva committed May 15, 2019
1 parent 1eba37a commit a9549b5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion ci/docker/install/tensorrt.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit a9549b5

Please sign in to comment.