cpu only

cmanzo · cmanzo · commit 91474be80c85 · 2025-05-16T17:29:16.000+02:00
diff --git a/docker/requirements.txt b/docker/requirements.txt
@@ -1,6 +1,7 @@
-torch==2.7.0
-torchvision==0.22.0
-torchaudio==2.7.0
+--index-url https://download.pytorch.org/whl/cpu
+torch==2.6.0
+torchvision==0.21.0
+torchaudio==2.6.0
 deeptrack
 deeplay
 pytorch-lightning
