Skip to content

Commit f24b687

Browse files
committed
Python TensorFlow 2.1 on Python 3.6 & CUDA10.1
1 parent 71e35ea commit f24b687

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

python-tensorflow/Dockerfile.2.1-py36-cuda101 renamed to python-tensorflow/Dockerfile.2.1-py36-cuda10.1

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ RUN apt-get update -y && \
9292
RUN find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
9393
rm /usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a
9494

95+
9596
RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 \
9697
&& echo "/usr/local/cuda/lib64/stubs" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
9798
&& ldconfig
@@ -322,7 +323,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
322323
RUN python3 -m pip install --no-cache-dir opencv-python && \
323324
rm -rf /root/.cache
324325

325-
# add 19.09.06
326+
# XGBoost
326327
RUN git clone --recursive https://github.com/dmlc/xgboost && \
327328
cd xgboost && \
328329
mkdir build && \
@@ -350,7 +351,7 @@ RUN mkdir /tmp/openmpi && \
350351
make -j $(nproc) all && \
351352
make install && \
352353
ldconfig && \
353-
rm -rf /tmp/openmp*i
354+
rm -rf /tmp/openmpi*
354355

355356
# Install Horovod, temporarily using CUDA stubs
356357
RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \

0 commit comments

Comments
 (0)