Update Dockerfile

Darveivoldavara · Jan 16, 2025 · beb6507 · beb6507
1 parent c96690e
commit beb6507
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.10-slim
+FROM python:3.12-slim
 
 WORKDIR /app
 
@@ -7,8 +7,8 @@ COPY /requirements.txt /app
 RUN apt-get update && apt-get install -y ffmpeg && rm -rf /var/lib/apt/lists/*
 RUN pip install \
     --default-timeout=100 \
-    torch==2.0.1 \
-    torchvision==0.15.2 -f https://download.pytorch.org/whl/cpu/torch_stable.html \
+    torch==2.2.0 \
+    torchvision==0.17.0 -f https://download.pytorch.org/whl/cpu/torch_stable.html \
     -r requirements.txt
 
 COPY /app /app