mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-13 23:53:02 +00:00
Brandon Thomas
85 lines
2.5 KiB
Docker
85 lines
2.5 KiB
Docker
# syntax=docker/dockerfile:1
|
|
|
|
FROM nvidia/cuda:12.0.1-runtime-ubuntu22.04
|
|
#FROM nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04
|
|
#FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# tzdata install hangs without presetting the time zone
|
|
ENV TZ=UTC
|
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
|
|
|
# NB: SadTalker requires ffmpeg
|
|
# NB: rsync to monitor time it takes for files to sync in k8s initContainer.
|
|
# NB: I had to remove `nvidia-driver-515` because it broke CUDA / nvidia-smi
|
|
# NB(bt,2025-01-20): StableAnimator uses git, git-lfs
|
|
# NB(bt,2025-01-20): unzip for web installs
|
|
# NB(bt,2025-01-20): debug / utility tools: lsof, htop, ripgrep, tmux, etc.
|
|
RUN apt-get update \
|
|
&& DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get install -y \
|
|
build-essential \
|
|
curl \
|
|
ffmpeg \
|
|
git \
|
|
git-lfs \
|
|
htop \
|
|
libsndfile1 \
|
|
lsof \
|
|
python3-pip \
|
|
python3.10 \
|
|
python3.10-dev \
|
|
python3.10-venv \
|
|
ripgrep \
|
|
rsync \
|
|
tmux \
|
|
unzip \
|
|
vim \
|
|
wget \
|
|
--no-install-recommends \
|
|
&& apt-get clean autoclean && apt-get autoremove -y && rm -rf /var/lib/{apt,dpkg,cache,log}/
|
|
|
|
# Install rclone
|
|
RUN cd /tmp && \
|
|
curl -O https://downloads.rclone.org/rclone-current-linux-amd64.zip && \
|
|
unzip rclone-current-linux-amd64.zip && \
|
|
cd rclone-*-linux-amd64 && \
|
|
cp rclone /usr/bin/ && \
|
|
chmod 755 /usr/bin/rclone && \
|
|
rm -rf /tmp/rclone*
|
|
|
|
# Install entr
|
|
RUN cd /tmp && \
|
|
wget -O entr.zip https://github.com/eradman/entr/archive/refs/tags/5.6.zip && \
|
|
unzip entr.zip && \
|
|
cd entr-5.6 && \
|
|
./configure && \
|
|
make test && \
|
|
make install \
|
|
&& rm -rf /tmp/entr-5.6
|
|
|
|
RUN mkdir /python_install
|
|
|
|
WORKDIR /python_install
|
|
|
|
COPY requirements.txt .
|
|
|
|
# NB: Kubernetes will copy over the venv we create here.
|
|
# NB: pip install taken from the StableAnimator README
|
|
RUN python3 -m venv python \
|
|
&& . python/bin/activate \
|
|
&& pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124 \
|
|
&& pip install torch==2.5.1+cu124 xformers --index-url https://download.pytorch.org/whl/cu124 \
|
|
&& pip install -r requirements.txt
|
|
|
|
# Model code is kept separate from the venv
|
|
RUN mkdir /model_code
|
|
|
|
WORKDIR /model_code
|
|
|
|
COPY . .
|
|
|
|
# NB(bt,2025-01-20): Let's do this outside of the image.
|
|
#RUN git lfs install
|
|
#RUN git clone https://huggingface.co/FrancisRing/StableAnimator checkpoints
|