sunjichen/enginex-ascend-910-diffusers
1
0
Fork 0
forked from EngineX-Ascend/enginex-ascend-910-diffusers
Code Pull Requests Activity

support mlu370

Browse Source
This commit is contained in:
ZHANG Hao
2025-08-25 17:07:12 +08:00
parent 4d35251531
commit b524f25741
2 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile.mlu370 Normal file
View File

@@ -0,0 +1,8 @@
FROM cambricon-base/pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310
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
RUN pip install diffusers==0.34.0
RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc
COPY main.py test.sh dataset.json /workspace/

4
run_in_docker_mlu370.sh Executable file
View File

@@ -0,0 +1,4 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:mlu370-0.2
device_id=7
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=/dev/cambricon_dev$device_id:/dev/cambricon_dev0 --device=/dev/cambricon_dev$device_id:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image bash