From 1e60b434cf7c65e9423b6a8daa8fb2f453680997 Mon Sep 17 00:00:00 2001 From: jkiesele Date: Fri, 27 Mar 2020 12:11:03 +0100 Subject: [PATCH] experimental docker file --- docker/Dockerfile | 8 ++++++- docker/Dockerfile_exp | 54 +++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 61 insertions(+), 1 deletion(-) create mode 100644 docker/Dockerfile_exp diff --git a/docker/Dockerfile b/docker/Dockerfile index 24d16f8..889b5f2 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,5 +1,5 @@ -FROM cernml4reco/djcbase:devpy3 +FROM cernml4reco/djcbase:py3 @@ -10,6 +10,12 @@ RUN export DEBIAN_FRONTEND=noninteractive && \ pip3 --no-cache-dir install tensorflow-gpu==2.1 +#2.1 +RUN cd /usr/local/lib/python3.6/dist-packages/tensorflow_core/include/third_party && \ + mkdir gpus && \ + cd gpus && \ + ln -s /usr/local/cuda cuda + #2.2.0-rc1 #RUN cd /usr/local/lib/python3.6/dist-packages/tensorflow/include/third_party && \ # mkdir gpus && \ diff --git a/docker/Dockerfile_exp b/docker/Dockerfile_exp new file mode 100644 index 0000000..079309a --- /dev/null +++ b/docker/Dockerfile_exp @@ -0,0 +1,54 @@ + +FROM cernml4reco/djcbase:py3 + + + +################################################################################ +# Tensorflow + +RUN export DEBIAN_FRONTEND=noninteractive && \ + pip3 --no-cache-dir install tensorflow-gpu==2.2.0-rc1 + + +#2.2.0-rc1 +RUN cd /usr/local/lib/python3.6/dist-packages/tensorflow/include/third_party && \ + mkdir gpus && \ + cd gpus && \ + ln -s /usr/local/cuda cuda + + + +################################################################################ +# the actual DeepJetCore. Make sure this gets built every time + +ARG BUILD_DATE +LABEL org.label-schema.build-date=$BUILD_DATE + +ENV DEEPJETCORE /usr/share/DJC/DeepJetCore + +RUN ldconfig && \ + cd /usr/share && \ + mkdir DJC && \ + cd DJC && \ + git clone https://github.com/DL4Jets/DeepJetCore && \ + cd DeepJetCore && \ +# git checkout 2.2_tf1.15 &&\ + source docker_env.sh && \ + cd compiled && \ + make -j4 + + +ENV PATH="/usr/share/DJC/DeepJetCore/bin:${PATH}" + +#/usr/local/lib is for root +ENV PYTHONPATH="/usr/share/DJC/DeepJetCore/../:${PYTHONPATH}" +ENV LD_LIBRARY_PATH="/usr/share/DJC/DeepJetCore/compiled:${LD_LIBRARY_PATH}" + + +# helpers for ragged and cuda compilation + + + +# The fix for TensorFlow + + \ No newline at end of file