docker run -it --device=/dev/vdavinci132:/dev/davinci132 --device=/dev/vdavinci133:/dev/davinci133 --device=/dev/vdavinci134:/dev/davinci134 --device=/dev/davinci_manager --device=/dev/devmm_svm --device=/dev/hisi_hdc -e  ASCEND_VISIBLE_DEVICES=132,133,134 -e ASCEND_RUNTIME_OPTIONS=VIRTUAL -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi  -v /usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/common -v /usr/local/Ascend/driver/lib64/driver:/usr/local/Ascend/driver/lib64/driver -d ascend-sam:v1 /bin/bash

​#!/usr/bin/env python3

import os
import sys
from ais_bench.infer.interface import InferSession

device_ids = [56, 132]

model_path = "weights/resnet18_image_encoder_310P3_FP32.om"
if len(sys.argv) > 1:
    model_path = sys.argv[1]

print(f"当前环境变量: ASCEND_VISIBLE_DEVICES={os.environ.get('ASCEND_VISIBLE_DEVICES', '未设置')}")
print(f"将尝试以下设备ID: {device_ids}")

for device_id in device_ids:
    print(f"\n尝试使用设备ID: {device_id}")
    try:
        os.environ["ASCEND_VISIBLE_DEVICES"] = str(device_id)
        print(f"设置ASCEND_VISIBLE_DEVICES={device_id}")
        
        session = InferSession(device_id, model_path)
        print(f"成功使用设备ID {device_id} 加载模型!")
        break
    except Exception as e:
        print(f"使用设备ID {device_id} 失败: {e}") 


​