镜像内已预置了适配昇腾平台的YOLO-World,仅需启动测试脚本,即可启动推理过程。具体操作步骤如下。
IMAGE_ID="your image ID"
container_name="your container name"
docker run \
--net=host \
--shm-size=512g \
--name $container_name \
--privileged \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro \
-v /usr/local/sbin:/usr/local/sbin:ro \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-it $IMAGE_ID bashcd /home/yolo_check
python yolo_world_xn.py