File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ FROM nvidia/cuda:11.3.0-cudnn8-devel-ubuntu20.04
2
2
3
3
# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
4
4
ARG CUDA=11.3
5
- ARG LIBNVINFER=8.0.0.3 -1
5
+ ARG LIBNVINFER=8.0.0-1
6
6
ARG LIBNVINFER_MAJOR_VERSION=8
7
7
ARG CUDNN=8.0.5.39-1
8
8
ENV NCCL=2.9.9
@@ -137,13 +137,15 @@ RUN rm -fr /var/lib/apt/lists/* && \
137
137
RUN curl -sL https://deb.nodesource.com/setup_14.x | bash - && \
138
138
apt-get update -y && \
139
139
apt-get install -y nodejs
140
-
140
+
141
141
RUN apt-get update && \
142
142
apt-get install -y --no-install-recommends \
143
143
libnvinfer${LIBNVINFER_MAJOR_VERSION}=${LIBNVINFER}+cuda${CUDA} \
144
144
libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
145
145
libnvinfer-plugin8 \
146
146
libnvinfer-plugin-dev=${LIBNVINFER}+cuda${CUDA} \
147
+ python3-libnvinfer=${LIBNVINFER}+cuda${CUDA} \
148
+ python3-libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
147
149
&& \
148
150
apt-get clean && \
149
151
rm -rf /var/lib/apt/lists/*
You can’t perform that action at this time.
0 commit comments