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

modify dockerfile

Browse Source
This commit is contained in:
ZHANG Hao
2025-09-03 15:55:09 +08:00
parent 722f25c905
commit eeda961724
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile
View File

@@ -3,6 +3,7 @@ FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0
WORKDIR /workspace WORKDIR /workspace
ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
RUN rm -rf /workspace/*
RUN pip install diffusers==0.34.0 RUN pip install diffusers==0.34.0
RUN pip install open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 RUN pip install open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0
RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc

1
run_in_docker_mlu370.sh
View File

@@ -1,5 +1,6 @@
#! /usr/bin/env bash #! /usr/bin/env bash
# cnmon # cnmon
image=harbor-contest.4pd.io/zhanghao/iic:mlu370 image=harbor-contest.4pd.io/zhanghao/iic:mlu370
image=harbor-contest.4pd.io/zhanghao/t2v:mlu370-0.1
device_id=2 device_id=2
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=/dev/cambricon_dev$device_id:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image bash docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=/dev/cambricon_dev$device_id:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image bash