add ascend support

This commit is contained in:
root
2025-09-03 10:16:24 +08:00
parent 33c57acb1b
commit 31c16831b1
4 changed files with 26 additions and 12 deletions

4
run_in_docker_ascend.sh Executable file
View File

@@ -0,0 +1,4 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:ascend-0.2
device=1
docker run -it -v `pwd`:/host -e ASCEND_VISIBLE_DEVICES=1 --device /dev/davinci$device:/dev/davinci0 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc -v /mnt:/mnt -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info -v /etc/ascend_install.info:/etc/ascend_install.info --privileged --entrypoint bash $image