diff --git a/Dockerfile b/Dockerfile
index d4cbd4f..968dd6f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -100,9 +100,9 @@ RUN set -x && \
 
 #RUN CONDA_CUDA_OVERRIDE="11.2" mamba install --yes -c conda-forge tensorflow==2.10.0=cuda112py310*
 ENV CONDA_CUDA_OVERRIDE="11.2"  
-RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
+#RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
 #==2.11.0=cuda*
-#RUN mamba install -y -c conda-forge tensorflow-gpu
+RUN mamba install -y -c conda-forge tensorflow-gpu
 WORKDIR /tmp
 RUN mamba install --quiet --yes \
     'notebook' \