File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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 && \
Original file line number Diff line number Diff line change 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+
You can’t perform that action at this time.
0 commit comments