File size: 2,265 Bytes
a2e3595 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | FROM nvidia/cuda:12.8.1-base-ubuntu22.04@sha256:001469ea0f3dec85a1ca929aeea3b58ae369d4c11228b10aec1f642bb6ca7a6f
ENV DEBIAN_FRONTEND=noninteractive
RUN rm -f /etc/apt/sources.list.d/cuda*.list \
&& sed -i 's|http://archive.ubuntu.com|https://archive.ubuntu.com|g; s|http://security.ubuntu.com|https://security.ubuntu.com|g' /etc/apt/sources.list \
&& apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 update \
&& apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 install -y --no-install-recommends \
ca-certificates libosmesa6 libgl1 libglu1-mesa libglib2.0-0 \
libsndfile1 libgomp1 libffi8 ffmpeg fontconfig fonts-dejavu-core \
&& rm -rf /var/lib/apt/lists/*
ADD environment.tar /opt/conda/envs/hybrid/
COPY environment-inventory.json requirements.snapshot.txt /opt/codiffact/
COPY smoke_test.py /opt/codiffact/smoke_test.py
RUN ln -sfn /usr/lib/x86_64-linux-gnu/libstdc++.so.6 /opt/conda/envs/hybrid/lib/libstdc++.so.6 \
&& ln -s /opt/conda/envs/hybrid/lib/python3.12/site-packages/imageio_ffmpeg/binaries/ffmpeg-linux64-v4.2.2 /usr/local/bin/ffmpeg \
&& mkdir -p /app /models /data /outputs /jizhi/jizhi2/worker/trainer/repo \
&& ln -s /app /jizhi/jizhi2/worker/trainer/repo/hybrid-motion
ENV PATH=/usr/local/bin:/opt/conda/envs/hybrid/bin:/usr/local/sbin:/usr/sbin:/usr/bin:/sbin:/bin \
PYOPENGL_PLATFORM=osmesa \
PYTHONNOUSERSITE=1 \
PYTHONDONTWRITEBYTECODE=1 \
HF_HOME=/models/hf_cache \
HF_HUB_OFFLINE=1 \
TOKENIZERS_PARALLELISM=false \
OMP_NUM_THREADS=4 \
MKL_NUM_THREADS=4 \
PYTHONPATH=/app/src:/app/third_party:/app/third_party/HY-Motion-1.0:/app
WORKDIR /app
RUN apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 update \
&& apt-get -o Acquire::ForceIPv4=true -o Acquire::Retries=2 -o Acquire::https::Timeout=45 install -y --no-install-recommends git git-lfs zip unzip \
&& rm -rf /var/lib/apt/lists/*
RUN python /opt/codiffact/smoke_test.py
LABEL org.opencontainers.image.title="CoDiffAct environment only" \
org.opencontainers.image.description="Python 3.12 / PyTorch 2.7.1 cu128; no project, pretrained weights, datasets or credentials"
CMD ["bash"]
|