Update Dockerfile

casus · Sep 27, 2024 · b0bf71d · b0bf71d
1 parent 3f6be33
commit b0bf71d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -38,7 +38,7 @@ RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
 # Set up micromamba environment, install dependencies and pip packages
 RUN micromamba create -n $ENV_NAME -y && \
     micromamba install -n $ENV_NAME \
-    tensorflow-gpu==2.15.0 \
+    tensorflow==2.15.0 \
     keras==2.15.* \
     matplotlib==3.8.0 \
     tqdm==4.65.0 \