diff --git a/Dockerfile.a100 b/Dockerfile.a100 new file mode 100644 index 0000000..a13584e --- /dev/null +++ b/Dockerfile.a100 @@ -0,0 +1,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 + +COPY . /workspace/ diff --git a/run_in_docker_a100.sh b/run_in_docker_a100.sh new file mode 100755 index 0000000..01291cb --- /dev/null +++ b/run_in_docker_a100.sh @@ -0,0 +1,3 @@ +#! /usr/bin/env bash +image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.1 +docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash