2.2 tf2.1

DL4Jets · Feb 28, 2020 · f76393a · f76393a
1 parent 0b58866
commit f76393a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,5 +1,5 @@
 
-FROM cernml4reco/djcbase:latest
+FROM cernml4reco/djcbase:py2
 
 RUN export DEBIAN_FRONTEND=noninteractive && \
     pip --no-cache-dir install tensorflow-gpu==2.1
@@ -34,7 +34,7 @@ RUN cd /usr/share && \
     cd DJC && \
     git clone https://github.com/DL4Jets/DeepJetCore && \
     cd DeepJetCore && \
-#    git checkout 2.2_tf1.15 &&\
+    git checkout 2.2_tf2.1 &&\
     source docker_env.sh && \
     cd compiled && \
     make -j4