storyforge-trainer (54d5e42bb022)

Published 2026-09-28 12:53:08 +02:00 by flying7eleven

Pull the image from the command line:

docker pull git.janke.biz/flying7eleven/storyforge-trainer:54d5e42bb022

Digest

sha256:cb514198f6773c7ae4206ae27be4cde61fc6feb10bc7dcd7fd1119eacfb1b515

For more information on the Container registry, see the documentation.

ARG RELEASE

ARG LAUNCHPAD_BUILD_ARCH

LABEL org.opencontainers.image.version=24.04

ADD file:0f6466425c4f1800aae9224ddc3437b90c829cea58fb8edd5dde2f1eb0ee28da in /

CMD ["/bin/bash"]

ENV NVARCH=x86_64

ENV NVIDIA_REQUIRE_CUDA=cuda>=13.0 brand=unknown,driver>=535,driver<536 brand=grid,driver>=535,driver<536 brand=tesla,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=vapps,driver>=535,driver<536 brand=vpc,driver>=535,driver<536 brand=vcs,driver>=535,driver<536 brand=vws,driver>=535,driver<536 brand=cloudgaming,driver>=535,driver<536 brand=unknown,driver>=550,driver<551 brand=grid,driver>=550,driver<551 brand=tesla,driver>=550,driver<551 brand=nvidia,driver>=550,driver<551 brand=quadro,driver>=550,driver<551 brand=quadrortx,driver>=550,driver<551 brand=nvidiartx,driver>=550,driver<551 brand=vapps,driver>=550,driver<551 brand=vpc,driver>=550,driver<551 brand=vcs,driver>=550,driver<551 brand=vws,driver>=550,driver<551 brand=cloudgaming,driver>=550,driver<551 brand=unknown,driver>=565,driver<566 brand=grid,driver>=565,driver<566 brand=tesla,driver>=565,driver<566 brand=nvidia,driver>=565,driver<566 brand=quadro,driver>=565,driver<566 brand=quadrortx,driver>=565,driver<566 brand=nvidiartx,driver>=565,driver<566 brand=vapps,driver>=565,driver<566 brand=vpc,driver>=565,driver<566 brand=vcs,driver>=565,driver<566 brand=vws,driver>=565,driver<566 brand=cloudgaming,driver>=565,driver<566 brand=unknown,driver>=570,driver<571 brand=grid,driver>=570,driver<571 brand=tesla,driver>=570,driver<571 brand=nvidia,driver>=570,driver<571 brand=quadro,driver>=570,driver<571 brand=quadrortx,driver>=570,driver<571 brand=nvidiartx,driver>=570,driver<571 brand=vapps,driver>=570,driver<571 brand=vpc,driver>=570,driver<571 brand=vcs,driver>=570,driver<571 brand=vws,driver>=570,driver<571 brand=cloudgaming,driver>=570,driver<571 brand=unknown,driver>=575,driver<576 brand=grid,driver>=575,driver<576 brand=tesla,driver>=575,driver<576 brand=nvidia,driver>=575,driver<576 brand=quadro,driver>=575,driver<576 brand=quadrortx,driver>=575,driver<576 brand=nvidiartx,driver>=575,driver<576 brand=vapps,driver>=575,driver<576 brand=vpc,driver>=575,driver<576 brand=vcs,driver>=575,driver<576 brand=vws,driver>=575,driver<576 brand=cloudgaming,driver>=575,driver<576

ENV NV_CUDA_CUDART_VERSION=13.0.96-1

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/${NVARCH}/cuda-keyring_1.1-1_all.deb && dpkg -i cuda-keyring_1.1-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit

ENV CUDA_VERSION=13.0.3

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-13-0=${NV_CUDA_CUDART_VERSION} cuda-toolkit-13-0-config-common=${NV_CUDA_CUDART_VERSION} cuda-toolkit-13-config-common=${NV_CUDA_CUDART_VERSION} cuda-toolkit-config-common=${NV_CUDA_CUDART_VERSION} && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && if apt-cache policy cuda-compat-13-0 2>/dev/null | grep -q "Candidate:"; then apt-get install -y --no-install-recommends cuda-compat-13-0; fi && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && if apt-cache policy cuda-compat-orin-13-0 2>/dev/null | grep -q "Candidate:"; then apt-get install -y --no-install-recommends cuda-compat-orin-13-0; fi && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64

COPY NGC-DL-CONTAINER-LICENSE / # buildkit

ENV NVIDIA_VISIBLE_DEVICES=all

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

ENV NV_CUDA_LIB_VERSION=13.0.3-1

ENV NV_NVTX_VERSION=13.0.85-1

ENV NV_LIBNPP_VERSION=13.0.1.2-1

ENV NV_LIBNPP_PACKAGE=libnpp-13-0=13.0.1.2-1

ENV NV_LIBCUSPARSE_VERSION=12.6.3.3-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-13-0

ENV NV_LIBCUBLAS_VERSION=13.1.1.3-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-13-0=13.1.1.3-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.28.3-1

ENV NCCL_VERSION=2.28.3-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.28.3-1+cuda13.0

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-libraries-13-0=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-13-0=${NV_NVTX_VERSION} libcusparse-13-0=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME} # buildkit

COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit

COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit

ENV NVIDIA_PRODUCT_NAME=CUDA

ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]

LABEL authors=jaret

ENV DEBIAN_FRONTEND=noninteractive

ENV TORCH_CUDA_ARCH_LIST=8.0 8.6 8.9 9.0 10.0 12.0

RUN /bin/sh -c apt-get update && apt-get install --no-install-recommends -y git curl build-essential cmake wget python3.12 python3-pip python3-dev python3-setuptools python3-wheel python3-venv ffmpeg tmux htop nvtop python3-opencv openssh-client openssh-server openssl rsync unzip && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit

WORKDIR /tmp

RUN /bin/sh -c curl -sL https://deb.nodesource.com/setup_23.x -o nodesource_setup.sh && bash nodesource_setup.sh && apt-get update && apt-get install -y nodejs && apt-get clean && rm -rf /var/lib/apt/lists/* # buildkit

WORKDIR /app

RUN /bin/sh -c ln -s /usr/bin/python3 /usr/bin/python # buildkit

RUN /bin/sh -c pip install --no-cache-dir torch==2.13.0 torchvision==0.28.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/cu130 --break-system-packages # buildkit

WORKDIR /app/ai-toolkit

COPY requirements.txt requirements_base.txt /app/ai-toolkit/ # buildkit

RUN /bin/sh -c pip install --no-cache-dir --break-system-packages -r requirements.txt && pip install setuptools==69.5.1 --no-cache-dir --break-system-packages # buildkit

RUN /bin/sh -c pip install --no-cache-dir --break-system-packages -U torchcodec==0.15.0 natten==0.21.7+torch2130cu130 --find-links https://whl.natten.org https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.47/flash_attn-2.8.3+cu130torch2.13-cp312-cp312-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl && python -c "import flash_attn, natten, torchcodec; print('accelerators OK:', flash_attn.__version__, natten.__version__, torchcodec.__version__)" # buildkit

COPY ui/package.json ui/package-lock.json /app/ai-toolkit/ui/ # buildkit

RUN /bin/sh -c cd /app/ai-toolkit/ui && npm ci # buildkit

ARG CACHEBUST=1790297293

ARG GIT_COMMIT=main

RUN |2 CACHEBUST=1790297293 GIT_COMMIT=main /bin/sh -c echo "Cache bust: ${CACHEBUST}" && git clone https://github.com/ostris/ai-toolkit.git /tmp/ai-toolkit-src && cd /tmp/ai-toolkit-src && git checkout ${GIT_COMMIT} && rsync -a --delete --exclude 'ui/node_modules' --exclude 'requirements.txt' --exclude 'ui/package.json' --exclude 'ui/package-lock.json' /tmp/ai-toolkit-src/ /app/ai-toolkit/ && rm -rf /tmp/ai-toolkit-src # buildkit

RUN |2 CACHEBUST=1790297293 GIT_COMMIT=main /bin/sh -c cd /app/ai-toolkit/ui && npm run update_db && npm run build # buildkit

EXPOSE [8675/tcp]

WORKDIR /

COPY docker/start.sh /start.sh # buildkit

RUN |2 CACHEBUST=1790297293 GIT_COMMIT=main /bin/sh -c chmod +x /start.sh # buildkit

CMD ["/start.sh"]

RUN /bin/sh -c pip install --no-cache-dir --break-system-packages fastapi==0.141.1 uvicorn==0.52.4 pyyaml # buildkit

WORKDIR /opt/storyforge

COPY prepare.py trainer.py entrypoint.sh /opt/storyforge/ # buildkit

RUN /bin/sh -c chmod +x /opt/storyforge/entrypoint.sh # buildkit

EXPOSE [8000/tcp]

CMD ["/opt/storyforge/entrypoint.sh"]

Key	Value
authors	jaret
maintainer	NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.version	24.04

Details

Container

2026-09-28 12:53:08 +02:00

OCI / Docker

linux/amd64

6.1 GiB

Versions (50) View all

latest

2026-10-06

3274d5dd35f3

2026-10-06

17708666782a

2026-10-06

4a9193fa5f32

2026-10-06

978e38daf2ba

2026-10-06

storyforge-trainer (54d5e42bb022)

Installation

Image layers

Labels