cpu only

Author: cmanzo

docker/Dockerfile

@@ -14,6 +14,11 @@ WORKDIR /home/jovyan/work
 COPY .. .
 
 # ---------- extra Python deps -----------------------------------------------------
+# CPU wheels only
+RUN pip install --no-cache-dir \
+        torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 \
+        --index-url https://download.pytorch.org/whl/cpu
+
 COPY docker/requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 

docker/requirements.txt

@@ -1,8 +1,4 @@
---index-url https://download.pytorch.org/whl/cpu
-torch==2.7.0
-torchvision==0.22.0
-torchaudio==2.7.0
-deeptrack==2.0.0
+deeptrack==2.0.1
 deeplay==0.1.3
 pytorch-lightning==2.5.1
 numpy==2.2.5