@@ -108,15 +108,15 @@ RUN rm -fr /var/lib/apt/lists/* && \
108
108
libgtk-3-dev \
109
109
libhdf5-dev \
110
110
libhdf5-serial-dev \
111
- libnvinfer8=${LIBNVINFER}+cuda${CUDA} \
111
+ libnvinfer8=${LIBNVINFER}+cuda${CUDA} \
112
112
libnvonnxparsers8=${LIBNVINFER}+cuda${CUDA} \
113
113
libnvparsers8=${LIBNVINFER}+cuda${CUDA} \
114
114
libnvinfer-plugin8=${LIBNVINFER}+cuda${CUDA} \
115
- libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
115
+ libnvinfer-dev=${LIBNVINFER}+cuda${CUDA} \
116
116
libnvonnxparsers-dev=${LIBNVINFER}+cuda${CUDA} \
117
117
libnvparsers-dev=${LIBNVINFER}+cuda${CUDA} \
118
118
libnvinfer-plugin-dev=${LIBNVINFER}+cuda${CUDA} \
119
- python3-libnvinfer=${LIBNVINFER}+cuda${CUDA} \
119
+ python3-libnvinfer=${LIBNVINFER}+cuda${CUDA} \
120
120
libjasper-dev \
121
121
libjpeg-dev \
122
122
liblapack-dev \
@@ -179,7 +179,7 @@ RUN rm -fr /var/lib/apt/lists/* && \
179
179
xz-utils \
180
180
yasm \
181
181
zip \
182
- zlib1g-dev && \
182
+ zlib1g-dev && \
183
183
# install nccl
184
184
curl -fsSL https://developer.download.nvidia.com/compute/redist/nccl/v2.11/nccl_2.11.4-1+cuda11.4_x86_64.txz -O && \
185
185
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.
422
422
423
423
# Clean up
424
424
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 && \
426
426
sed -i 's/source \/ usr\/ local\/ nvm\/ nvm.sh//' /etc/bash.bashrc && \
427
427
rm -rf /var/lib/apt/lists/* && \
428
428
rm -rf /root/.cache && \
0 commit comments