Skip to content

Commit 1e60b43

Browse files
committed
experimental docker file
1 parent eb1438f commit 1e60b43

2 files changed

Lines changed: 61 additions & 1 deletion

File tree

docker/Dockerfile

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11

2-
FROM cernml4reco/djcbase:devpy3
2+
FROM cernml4reco/djcbase:py3
33

44

55

@@ -10,6 +10,12 @@ RUN export DEBIAN_FRONTEND=noninteractive && \
1010
pip3 --no-cache-dir install tensorflow-gpu==2.1
1111

1212

13+
#2.1
14+
RUN cd /usr/local/lib/python3.6/dist-packages/tensorflow_core/include/third_party && \
15+
mkdir gpus && \
16+
cd gpus && \
17+
ln -s /usr/local/cuda cuda
18+
1319
#2.2.0-rc1
1420
#RUN cd /usr/local/lib/python3.6/dist-packages/tensorflow/include/third_party && \
1521
# mkdir gpus && \

docker/Dockerfile_exp

Lines changed: 54 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
1+
2+
FROM cernml4reco/djcbase:py3
3+
4+
5+
6+
################################################################################
7+
# Tensorflow
8+
9+
RUN export DEBIAN_FRONTEND=noninteractive && \
10+
pip3 --no-cache-dir install tensorflow-gpu==2.2.0-rc1
11+
12+
13+
#2.2.0-rc1
14+
RUN cd /usr/local/lib/python3.6/dist-packages/tensorflow/include/third_party && \
15+
mkdir gpus && \
16+
cd gpus && \
17+
ln -s /usr/local/cuda cuda
18+
19+
20+
21+
################################################################################
22+
# the actual DeepJetCore. Make sure this gets built every time
23+
24+
ARG BUILD_DATE
25+
LABEL org.label-schema.build-date=$BUILD_DATE
26+
27+
ENV DEEPJETCORE /usr/share/DJC/DeepJetCore
28+
29+
RUN ldconfig && \
30+
cd /usr/share && \
31+
mkdir DJC && \
32+
cd DJC && \
33+
git clone https://github.com/DL4Jets/DeepJetCore && \
34+
cd DeepJetCore && \
35+
# git checkout 2.2_tf1.15 &&\
36+
source docker_env.sh && \
37+
cd compiled && \
38+
make -j4
39+
40+
41+
ENV PATH="/usr/share/DJC/DeepJetCore/bin:${PATH}"
42+
43+
#/usr/local/lib is for root
44+
ENV PYTHONPATH="/usr/share/DJC/DeepJetCore/../:${PYTHONPATH}"
45+
ENV LD_LIBRARY_PATH="/usr/share/DJC/DeepJetCore/compiled:${LD_LIBRARY_PATH}"
46+
47+
48+
# helpers for ragged and cuda compilation
49+
50+
51+
52+
# The fix for TensorFlow
53+
54+

0 commit comments

Comments
 (0)