diff --git a/Dockerfile b/Dockerfile
index 5137800..6526a28 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -99,5 +99,6 @@ RUN set -x && \
 #RUN conda install -c conda-forge tensorflow-gpu=2.11 -y 
 
 #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 -c conda-forge tensorflow-gpu
+ENV CONDA_CUDA_OVERRIDE="11.2"  
+RUN mamba install -y tensorflow cudatoolkit>=11.2 -c conda-forge
+#RUN mamba install -y -c conda-forge tensorflow-gpu