File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -92,6 +92,7 @@ RUN apt-get update -y && \
92
92
RUN find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
93
93
rm /usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a
94
94
95
+
95
96
RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 \
96
97
&& echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
97
98
&& ldconfig
@@ -322,7 +323,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
322
323
RUN python3 -m pip install --no-cache-dir opencv-python && \
323
324
rm -rf /root/.cache
324
325
325
- # add 19.09.06
326
+ # XGBoost
326
327
RUN git clone --recursive https://github.com/dmlc/xgboost && \
327
328
cd xgboost && \
328
329
mkdir build && \
@@ -350,7 +351,7 @@ RUN mkdir /tmp/openmpi && \
350
351
make -j $(nproc) all && \
351
352
make install && \
352
353
ldconfig && \
353
- rm -rf /tmp/openmp*i
354
+ rm -rf /tmp/openmpi*
354
355
355
356
# Install Horovod, temporarily using CUDA stubs
356
357
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
You can’t perform that action at this time.
0 commit comments