diff --git a/run_in_docker_mlu370.sh b/run_in_docker_mlu370.sh index 0bf7717..b76b3df 100755 --- a/run_in_docker_mlu370.sh +++ b/run_in_docker_mlu370.sh @@ -1,4 +1,5 @@ #! /usr/bin/env bash +# cnmon 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 +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