diff --git a/docker/build/x86_64/Dockerfile-CUDA b/docker/build/x86_64/Dockerfile-CUDA index 616e83a42..bdd82a18f 100644 --- a/docker/build/x86_64/Dockerfile-CUDA +++ b/docker/build/x86_64/Dockerfile-CUDA @@ -1,4 +1,5 @@ # This dockerfile should be built with `make docker-cuda-build` from the stash root. +ARG CUDA_VERSION=12.8.0 # Build Frontend FROM node:20-alpine AS frontend @@ -34,19 +35,26 @@ ARG STASH_VERSION RUN make flags-release flags-pie stash # Final Runnable Image -FROM nvidia/cuda:12.0.1-base-ubuntu22.04 -RUN apt update && apt upgrade -y && apt install -y ca-certificates libvips-tools ffmpeg wget intel-media-va-driver-non-free vainfo -RUN rm -rf /var/lib/apt/lists/* -COPY --from=backend /stash/stash /usr/bin/ +FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu24.04 +RUN apt update && apt upgrade -y && apt install -y \ + # stash dependencies + ca-certificates libvips-tools ffmpeg \ + # intel dependencies + intel-media-va-driver-non-free vainfo \ + # python tools + python3 python3-pip && \ + # cleanup + apt autoremove -y && apt clean && \ + rm -rf /var/lib/apt/lists/* +COPY --from=backend --chmod=555 /stash/stash /usr/bin/ # NVENC Patch RUN mkdir -p /usr/local/bin /patched-lib -RUN wget https://raw.githubusercontent.com/keylase/nvidia-patch/master/patch.sh -O /usr/local/bin/patch.sh -RUN wget https://raw.githubusercontent.com/keylase/nvidia-patch/master/docker-entrypoint.sh -O /usr/local/bin/docker-entrypoint.sh -RUN chmod +x /usr/local/bin/patch.sh /usr/local/bin/docker-entrypoint.sh /usr/bin/stash +ADD --chmod=555 https://raw.githubusercontent.com/keylase/nvidia-patch/master/patch.sh /usr/local/bin/patch.sh +ADD --chmod=555 https://raw.githubusercontent.com/keylase/nvidia-patch/master/docker-entrypoint.sh /usr/local/bin/docker-entrypoint.sh -ENV LANG C.UTF-8 -ENV NVIDIA_VISIBLE_DEVICES all +ENV LANG=C.UTF-8 +ENV NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_DRIVER_CAPABILITIES=video,utility ENV STASH_CONFIG_FILE=/root/.stash/config.yml EXPOSE 9999