From d5d20fd29f8b8c5dd2f7d27219a373cdf56da397 Mon Sep 17 00:00:00 2001 From: "google-labs-jules[bot]" <161369871+google-labs-jules[bot]@users.noreply.github.com> Date: Sun, 4 Jan 2026 17:55:48 +0000 Subject: [PATCH] Enable GPU support in Docker container - Modified `Dockerfile` to patch `backend/requirements.txt` during build. - Replaced `torch==1.12.0+cpu` with `torch==1.12.0+cu113` to install CUDA-enabled PyTorch. - This ensures GPU availability in the container environment while keeping the local `requirements.txt` optimized for CPU/Dev environments. --- Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile b/Dockerfile index 040f325..79c04dc 100644 --- a/Dockerfile +++ b/Dockerfile @@ -22,6 +22,8 @@ ENV LISTEN_PORT 80 #install dependencies COPY ./backend/requirements.txt /app +# Enable GPU support by using CUDA version of PyTorch +RUN sed -i 's/torch==1.12.0+cpu/torch==1.12.0+cu113/g' /app/requirements.txt RUN pip install -r /app/requirements.txt COPY ./backend/requirements_aligner.txt /app