bge embedding support

This commit is contained in:
2025-09-03 10:43:22 +08:00
commit f0425d03b4
4 changed files with 104 additions and 0 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/feature:ascend-0.1
device=1
docker run -it -v `pwd`:/host -e ASCEND_VISIBLE_DEVICES=$device -e NPU_VISIBLE_DEVICES=${device} --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