From 5e2769ecd6754525f72b18b8ead0443b2af7776a Mon Sep 17 00:00:00 2001 From: ZHANG Hao Date: Mon, 8 Sep 2025 17:33:05 +0800 Subject: [PATCH] update a100 image --- Dockerfile.a100 | 3 ++- run_in_docker_a100.sh | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile.a100 b/Dockerfile.a100 index a13584e..852ec90 100644 --- a/Dockerfile.a100 +++ b/Dockerfile.a100 @@ -2,6 +2,7 @@ FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel WORKDIR /workspace -RUN pip install opencv-python-headless imageio[ffmpeg] einops datasets==3.2.0 simplejson diffusers==0.34.0 open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict -i https://nexus.4pd.io/repository/pypi-all/simple +RUN pip install pytorch-lightning opencv-python-headless imageio[ffmpeg] einops datasets==3.2.0 simplejson diffusers==0.34.0 open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict -i https://nexus.4pd.io/repository/pypi-all/simple +RUN pip install transformers accelerate -i https://nexus.4pd.io/repository/pypi-all/simple COPY . /workspace/ diff --git a/run_in_docker_a100.sh b/run_in_docker_a100.sh index 01291cb..8919057 100755 --- a/run_in_docker_a100.sh +++ b/run_in_docker_a100.sh @@ -1,3 +1,3 @@ #! /usr/bin/env bash -image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.1 +image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.2 docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash