Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions container-images/cuda/Containerfile
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
# Base image with CUDA for compilation
FROM docker.io/nvidia/cuda:12.6.0-devel-ubi9 AS builder
FROM docker.io/nvidia/cuda:12.8.0-devel-ubi9 AS builder

COPY ../scripts /scripts
RUN chmod +x /scripts/*.sh && \
/scripts/build_llama_and_whisper.sh "cuda"

# Final runtime image
FROM docker.io/nvidia/cuda:12.6.0-runtime-ubi9
FROM docker.io/nvidia/cuda:12.8.0-runtime-ubi9

RUN dnf install -y python3 && \
dnf clean all && rm -rf /var/cache/*dnf*
Expand Down
Loading