| 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"] |
|
|
