Skip to content

Commit 232401c

Browse files
committed
ci: add GitHub Actions workflow to build and push Docker images
1 parent 1969df2 commit 232401c

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

Dockerfile.cuda

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@ ENV PATH="/app/venv/bin:$PATH"
1616

1717
RUN pip install --no-cache-dir torch torchvision torchaudio \
1818
--index-url https://download.pytorch.org/whl/cu129 \
19-
FROM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu24.04 AS runtime
19+
&& pip install --no-cache-dir -r requirements.txt
20+
FROM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu24.04 AS runtime
2021

2122
ARG DEBIAN_FRONTEND=noninteractive
2223

0 commit comments

Comments
 (0)