From 0ae0bd79044dcb0f81d9a38247d472f2d97ba36f Mon Sep 17 00:00:00 2001 From: "yuxiaojie@4paradigm.com" Date: Mon, 8 Sep 2025 16:29:07 +0800 Subject: [PATCH] support kunlunxin --- Dockerfile.r200 | 8 ++++++++ run_in_docker_r200.sh | 5 +++++ 2 files changed, 13 insertions(+) create mode 100644 Dockerfile.r200 create mode 100755 run_in_docker_r200.sh diff --git a/Dockerfile.r200 b/Dockerfile.r200 new file mode 100644 index 0000000..f65b1a1 --- /dev/null +++ b/Dockerfile.r200 @@ -0,0 +1,8 @@ +FROM git.modelhub.org.cn:9443/enginex-kunlunxin/text2image/r200_8f-diffuser:v0.21.4 + +WORKDIR /workspace + +RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && pip install imageio[ffmpeg] einops datasets==3.1.0 simplejson addict sortedcontainers modelscope==1.28.2 av==11.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple +RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && pip install megatron megatron-core urllib3==1.26.20 -i https://pypi.tuna.tsinghua.edu.cn/simple +RUN source /root/miniconda/etc/profile.d/conda.sh && conda activate python38_torch201_cuda && pip install open_clip_torch==2.24.0 pytorch-lightning==2.0.1 -i https://pypi.tuna.tsinghua.edu.cn/simple +COPY . /workspace/ diff --git a/run_in_docker_r200.sh b/run_in_docker_r200.sh new file mode 100755 index 0000000..183df3d --- /dev/null +++ b/run_in_docker_r200.sh @@ -0,0 +1,5 @@ +#! /usr/bin/env bash +# cnmon +image=harbor-contest.4pd.io/zhanghao/t2v:r200-0.1 +device_id=2 +docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --security-opt=seccomp=unconfined --cap-add=SYS_PTRACE --cap-add=SYS_ADMIN --device /dev/fuse --shm-size=32g --ulimit=memlock=-1 --ulimit=nofile=120000 --ulimit=stack=67108864 --device=/dev/xpu$device_id:/dev/xpu0 --device=/dev/xpuctrl:/dev/xpuctrl $image bash