Skip to content

Commit

Permalink
experimental docker file
Browse files Browse the repository at this point in the history
  • Loading branch information
jkiesele committed Mar 27, 2020
1 parent eb1438f commit 1e60b43
Show file tree
Hide file tree
Showing 2 changed files with 61 additions and 1 deletion.
8 changes: 7 additions & 1 deletion docker/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@

FROM cernml4reco/djcbase:devpy3
FROM cernml4reco/djcbase:py3



Expand All @@ -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 && \
Expand Down
54 changes: 54 additions & 0 deletions docker/Dockerfile_exp
Original file line number Diff line number Diff line change
@@ -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


0 comments on commit 1e60b43

Please sign in to comment.