From b0bf71d3e90be227cbe64b89c86cc0fedd51c29f Mon Sep 17 00:00:00 2001
From: Maria Wyrzykowska <maryswyrz@gmail.com>
Date: Fri, 27 Sep 2024 11:38:42 +0200
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 02b8121..55bd231 100644
--- 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 \