From b85c56f5ade2993525aabf24c5fcbd117ab5152b Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Tue, 13 Jan 2026 23:59:57 -0800
Subject: [PATCH] Try avoiding cuda deps

---
 Dockerfile | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 5262ea124..bd72265a0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -195,8 +195,20 @@ RUN set -eux \
     && apt-get update \
     && apt-get install --yes --quiet --no-install-recommends ${BUILD_PACKAGES} \
   && echo "Installing Python requirements" \
+    # Ensure linux/amd64 uses CPU-only PyTorch wheels (avoid pulling CUDA/nvidia-* deps) \
+    && printf '%s\n' \
+         '--extra-index-url https://download.pytorch.org/whl/cpu' \
+         'torch==2.7.0' \
+         > /tmp/torch-cpu-constraints.txt \
     && uv export --quiet --no-dev --all-extras --format requirements-txt --output-file requirements.txt \
-    && uv pip install --no-cache --system --no-python-downloads --python-preference system --requirements requirements.txt \
+    && if [ "${TARGETARCH}" = "amd64" ]; then \
+         uv pip install --no-cache --system --no-python-downloads --python-preference system \
+           -c /tmp/torch-cpu-constraints.txt \
+           --requirements requirements.txt ; \
+       else \
+         uv pip install --no-cache --system --no-python-downloads --python-preference system \
+           --requirements requirements.txt ; \
+       fi \
   && echo "Installing NLTK data" \
     && python3 -W ignore::RuntimeWarning -m nltk.downloader -d "/usr/share/nltk_data" snowball_data \
     && python3 -W ignore::RuntimeWarning -m nltk.downloader -d "/usr/share/nltk_data" stopwords \