diff --git a/Dockerfile b/Dockerfile
index 810958e..e9ab0f4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -100,7 +100,7 @@ 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-gpu cudatoolkit>=11.2 -c conda-forge
 #RUN mamba install -y -c conda-forge tensorflow-gpu
 WORKDIR /tmp
 RUN mamba install --quiet --yes \