From b524f25741b79433a6fd588c0158534ed965c066 Mon Sep 17 00:00:00 2001
From: ZHANG Hao <zhanghao@4paradigm.com>
Date: Mon, 25 Aug 2025 17:07:12 +0800
Subject: [PATCH] support mlu370

---
 Dockerfile.mlu370       | 8 ++++++++
 run_in_docker_mlu370.sh | 4 ++++
 2 files changed, 12 insertions(+)
 create mode 100644 Dockerfile.mlu370
 create mode 100755 run_in_docker_mlu370.sh

diff --git a/Dockerfile.mlu370 b/Dockerfile.mlu370
new file mode 100644
index 0000000..d8a5a55
--- /dev/null
+++ b/Dockerfile.mlu370
@@ -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/
diff --git a/run_in_docker_mlu370.sh b/run_in_docker_mlu370.sh
new file mode 100755
index 0000000..0bf7717
--- /dev/null
+++ b/run_in_docker_mlu370.sh
@@ -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