Skip to content

Commit b4554b5

Browse files
authored
Update Dockerfile.base.py38-cuda11.5
1 parent 2f7c321 commit b4554b5

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

commons/Dockerfile.base.py38-cuda11.5

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -108,15 +108,15 @@ RUN rm -fr /var/lib/apt/lists/* && \
108108
libgtk-3-dev \
109109
libhdf5-dev \
110110
libhdf5-serial-dev \
111-
libnvinfer8=${LIBNVINFER}+cuda${CUDA} \
111+
libnvinfer8=${LIBNVINFER}+cuda${CUDA} \
112112
libnvonnxparsers8=${LIBNVINFER}+cuda${CUDA} \
113113
libnvparsers8=${LIBNVINFER}+cuda${CUDA} \
114114
libnvinfer-plugin8=${LIBNVINFER}+cuda${CUDA} \
115-
libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
115+
libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
116116
libnvonnxparsers-dev=${LIBNVINFER}+cuda${CUDA} \
117117
libnvparsers-dev=${LIBNVINFER}+cuda${CUDA} \
118118
libnvinfer-plugin-dev=${LIBNVINFER}+cuda${CUDA} \
119-
python3-libnvinfer=${LIBNVINFER}+cuda${CUDA} \
119+
python3-libnvinfer=${LIBNVINFER}+cuda${CUDA} \
120120
libjasper-dev \
121121
libjpeg-dev \
122122
liblapack-dev \
@@ -179,7 +179,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
179179
xz-utils \
180180
yasm \
181181
zip \
182-
zlib1g-dev && \
182+
zlib1g-dev && \
183183
# install nccl
184184
curl -fsSL https://developer.download.nvidia.com/compute/redist/nccl/v2.11/nccl_2.11.4-1+cuda11.4_x86_64.txz -O && \
185185
tar --no-same-owner --keep-old-files --lzma -xvf nccl_${NCCL}-1+cuda11.4_x86_64.txz -C /usr/local/cuda/lib64/ --strip-components=2 --wildcards '*/lib/libnccl.so.*' && \
@@ -422,7 +422,7 @@ RUN echo "btl_tcp_if_exclude = lo,docker0" >> /usr/local/etc/openmpi-mca-params.
422422

423423
# Clean up
424424
RUN apt autoclean && \
425-
ln -s /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcusolver.so.11 /usr/local/cuda-11.4/targets/x86_64-linux/lib/libcusolver.so.10 && \
425+
ln -s /usr/local/cuda-11.5/targets/x86_64-linux/lib/libcusolver.so.11 /usr/local/cuda-11.5/targets/x86_64-linux/lib/libcusolver.so.10 && \
426426
sed -i 's/source \/usr\/local\/nvm\/nvm.sh//' /etc/bash.bashrc && \
427427
rm -rf /var/lib/apt/lists/* && \
428428
rm -rf /root/.cache && \

0 commit comments

Comments
 (0)