We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1969df2 commit 232401cCopy full SHA for 232401c
1 file changed
Dockerfile.cuda
@@ -16,7 +16,8 @@ ENV PATH="/app/venv/bin:$PATH"
16
17
RUN pip install --no-cache-dir torch torchvision torchaudio \
18
--index-url https://download.pytorch.org/whl/cu129 \
19
- FROM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu24.04 AS runtime
+ && pip install --no-cache-dir -r requirements.txt
20
+FROM nvidia/cuda:12.9.2-cudnn-runtime-ubuntu24.04 AS runtime
21
22
ARG DEBIAN_FRONTEND=noninteractive
23
0 commit comments