Skip to content

Commit b79a9d9

Browse files
committed
ci: add GitHub Actions workflow to build and push Docker images
1 parent e1ff117 commit b79a9d9

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

Dockerfile.cuda

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,7 @@ 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-
&& pip install --no-cache-dir -r requirements.txt
20-
FROM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu24.04 AS runtime
19+
FROM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu24.04 AS runtime
2120

2221
ARG DEBIAN_FRONTEND=noninteractive
2322

@@ -36,4 +35,4 @@ ENV PYTHONUNBUFFERED=1
3635
ENV NVIDIA_VISIBLE_DEVICES=all
3736
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
3837

39-
ENTRYPOINT ["python3", "engine/main.py"] # ← change to your entrypoint
38+
ENTRYPOINT ["python3", "engine/main.py"]

0 commit comments

Comments
 (0)