support mlu370

2025-08-25 17:07:12 +08:00
parent 4d35251531
commit b524f25741
2 changed files with 12 additions and 0 deletions
--- a/run_in_docker_mlu370.sh
+++ 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