EngineX-Ascend/enginex-ascend-910-text2video
10
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

support kunlunxin

Browse Source
This commit is contained in:
yuxiaojie@4paradigm.com
2025-09-08 16:29:07 +08:00
parent 08b5a1433a
commit 0ae0bd7904
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile.r200 Normal file
View File

@@ -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/

5
run_in_docker_r200.sh Executable file
View File

@@ -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