From 1b590aad758b201c97f7e82fdc913bd795c7d643 Mon Sep 17 00:00:00 2001
From: ZHANG Hao <zhanghao@4paradigm.com>
Date: Wed, 3 Sep 2025 10:51:33 +0800
Subject: [PATCH] dockerfile for a100

---
 Dockerfile.a100       | 7 +++++++
 run_in_docker_a100.sh | 3 +++
 2 files changed, 10 insertions(+)
 create mode 100644 Dockerfile.a100
 create mode 100755 run_in_docker_a100.sh

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