diff --git a/docker/Dockerfile b/docker/Dockerfile
index 24d16f8..889b5f2 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 
-FROM cernml4reco/djcbase:devpy3
+FROM cernml4reco/djcbase:py3
 
 
 
@@ -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 && \
diff --git a/docker/Dockerfile_exp b/docker/Dockerfile_exp
new file mode 100644
index 0000000..079309a
--- /dev/null
+++ b/docker/Dockerfile_exp
@@ -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
+
+    
\ No newline at end of file