TF1.15 tag

DL4Jets · Feb 28, 2020 · 02b2951 · 02b2951
1 parent f76393a
commit 02b2951
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -2,7 +2,7 @@
 FROM cernml4reco/djcbase:py2
 
 RUN export DEBIAN_FRONTEND=noninteractive && \
-    pip --no-cache-dir install tensorflow-gpu==2.1
+    pip --no-cache-dir install tensorflow-gpu==1.15
 
 RUN cd /usr/local/lib/python2.7/dist-packages/tensorflow_core/include/third_party && \
     mkdir gpus && \
@@ -34,7 +34,7 @@ RUN cd /usr/share && \
     cd DJC && \
     git clone https://github.com/DL4Jets/DeepJetCore && \
     cd DeepJetCore && \
-    git checkout 2.2_tf2.1 &&\
+    git checkout 2.2_tf1.15 &&\
     source docker_env.sh && \
     cd compiled && \
     make -j4