Skip to content

Commit 5b33c60

Browse files
fix the docker build failure on main (#2942)
1 parent 072c34e commit 5b33c60

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

docker/Dockerfile

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,15 @@ RUN pyenv global ${PYTHON_VERSION}
3636
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
3737
RUN add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
3838
RUN apt-get update
39-
RUN TENSORRT_MAJOR_VERSION=`echo ${TENSORRT_VERSION} | cut -d '.' -f 1` && apt-get install -y libnvinfer${TENSORRT_MAJOR_VERSION}=${TENSORRT_VERSION}.* libnvinfer-plugin${TENSORRT_MAJOR_VERSION}=${TENSORRT_VERSION}.* libnvinfer-dev=${TENSORRT_VERSION}.* libnvinfer-plugin-dev=${TENSORRT_VERSION}.* libnvonnxparsers${TENSORRT_MAJOR_VERSION}=${TENSORRT_VERSION}.* libnvonnxparsers-dev=${TENSORRT_VERSION}.*
39+
RUN TENSORRT_MAJOR_VERSION=`echo ${TENSORRT_VERSION} | cut -d '.' -f 1` && \
40+
apt-get install -y libnvinfer${TENSORRT_MAJOR_VERSION}=${TENSORRT_VERSION}.* \
41+
libnvinfer-plugin${TENSORRT_MAJOR_VERSION}=${TENSORRT_VERSION}.* \
42+
libnvinfer-dev=${TENSORRT_VERSION}.* \
43+
libnvinfer-headers-dev=${TENSORRT_VERSION}.* \
44+
libnvinfer-headers-plugin-dev=${TENSORRT_VERSION}.* \
45+
libnvinfer-plugin-dev=${TENSORRT_VERSION}.* \
46+
libnvonnxparsers${TENSORRT_MAJOR_VERSION}=${TENSORRT_VERSION}.* \
47+
libnvonnxparsers-dev=${TENSORRT_VERSION}.*
4048

4149
# Setup Bazel via Bazelisk
4250
RUN wget -q https://github.com/bazelbuild/bazelisk/releases/download/v1.17.0/bazelisk-linux-amd64 -O /usr/bin/bazel &&\

0 commit comments

Comments
 (0)