Fix torch related compatibility issues

Surveily · Nov 20, 2024 · e31eccd · e31eccd
1 parent 43aa778
commit e31eccd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/surveily.developer.ai/base/Dockerfile b/src/surveily.developer.ai/base/Dockerfile
@@ -53,9 +53,9 @@ USER vscode
 # - Runtime parameters
 RUN python3 -m pip install -U pip && \
     python3 -m pip install --user --quiet --no-cache-dir \
-    torch==2.0.0 \
-    torchvision==0.15.1 \
-    ultralytics==8.2.28 \
+    torch==2.5.1 \
+    torchvision==0.20.1 \
+    ultralytics==8.3.34 \
     huggingface_hub==0.18.0
 
 RUN python3 -m pip install --user --quiet --no-cache-dir \
@@ -90,7 +90,7 @@ RUN python3 -m pip install --user --quiet --no-cache-dir \
     nbconvert \
     xmltodict \
     autoflake \
-    lightning==2.3.3 \
+    lightning==2.4.0 \
     streamlit \
     pytest-cov \
     ipywidgets \