Skip to content

Commit 51f710c

Browse files
committed
NVinfer 8.0.0.3
1 parent da5dd70 commit 51f710c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

commons/Dockerfile.base.py38-cuda11.3

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ FROM nvidia/cuda:11.3.0-cudnn8-devel-ubuntu20.04
22

33
# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
44
ARG CUDA=11.3
5-
ARG LIBNVINFER=8.0.0.3-1
5+
ARG LIBNVINFER=8.0.0-1
66
ARG LIBNVINFER_MAJOR_VERSION=8
77
ARG CUDNN=8.0.5.39-1
88
ENV NCCL=2.9.9
@@ -137,13 +137,15 @@ RUN rm -fr /var/lib/apt/lists/* && \
137137
RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
138138
apt-get update -y && \
139139
apt-get install -y nodejs
140-
140+
141141
RUN apt-get update && \
142142
apt-get install -y --no-install-recommends \
143143
libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
144144
libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
145145
libnvinfer-plugin8 \
146146
libnvinfer-plugin-dev=${LIBNVINFER}+cuda${CUDA} \
147+
python3-libnvinfer=${LIBNVINFER}+cuda${CUDA} \
148+
python3-libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
147149
&& \
148150
apt-get clean && \
149151
rm -rf /var/lib/apt/lists/*

0 commit comments

Comments
 (0)