📌 Use cpu torch

Author: Paillat-dev

Dockerfile

@@ -6,7 +6,9 @@ RUN apt-get update && apt-get install -y curl build-essential
 ENV PYTHONUNBUFFERED 1
 ENV PYTHONDONTWRITEBYTECODE 1
 ADD requirements.txt /app/requirements.txt
-RUN pip install -r requirements.txt
+RUN export TORCH_VERSION=$(grep torch== requirements.txt | cut -d'=' -f3) && \
+    pip install torch==$TORCH_VERSION --index-url https://download.pytorch.org/whl/cpu --no-cache-dir && \
+    pip install -r requirements.txt --no-cache-dir
 
 # ADDING FILES
 ADD app.py /app/app.py

pdm.lock

@@ -31,3 +31,16 @@ dev = [
 
 [tool.pdm]
 distribution = false
+
+[[tool.pdm.source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+exclude_packages = ["torch", "torchvision", "torchaudio"]
+
+[[tool.pdm.source]]
+name = "torch"
+url = "https://download.pytorch.org/whl/cpu"
+include_packages = ["torch", "torchvision", "torchaudio"]
+exclude_packages = ["*"]
+

pyproject.toml

@@ -60,7 +60,7 @@ shellingham==1.5.4
 sniffio==1.3.1
 starlette==0.37.2
 sympy==1.12.1
-tbb==2021.12.0; platform_system == "Windows"
+tbb==2021.13.0; platform_system == "Windows"
 tokenizers==0.19.1
 torch==2.3.1
 tqdm==4.66.4
@@ -75,3 +75,5 @@ uvloop==0.19.0; (sys_platform != "cygwin" and sys_platform != "win32") and platf
 watchdog==4.0.1
 watchfiles==0.22.0
 websockets==12.0
+--index-url https://pypi.org/simple
+--extra-index-url https://download.pytorch.org/whl/cpu