YOLOv8 CLI预测时，如何指定参数使用Intel OpenCL GPU？

问题说明 #

已安装OpenVINO依赖并将模型转换为OpenVINO格式，系统存在可用OpenCL设备：

$ clinfo -l
Platform #0: Intel(R) OpenCL HD Graphics
 `-- Device #0: Intel(R) Graphics [0xa7a0]

执行yolo predict时，使用device=gpu或device=0会报错Invalid CUDA device requested，但CPU运行正常。

解决方案 #

YOLO默认的device=gpu参数指向CUDA设备，要调用Intel的OpenCL GPU，需明确指定OpenVINO后端的设备类型，使用以下格式的命令：

• 调用Intel GPU（通用写法）：

yolo predict model=openvino_model source='samples/*.jpg' device=openvino:GPU

• 指定具体设备索引（对应clinfo中的Device #0）：

yolo predict model=openvino_model source='samples/*.jpg' device=openvino:0

原理 #

当直接使用device=gpu时，YOLO会尝试寻找CUDA兼容的NVIDIA GPU，而你的环境中只有Intel的OpenCL GPU，因此触发错误。通过openvino:GPU或openvino:设备索引的格式，可让YOLO通过OpenVINO后端调用Intel的OpenCL设备。

内容的提问来源于stack exchange，提问作者Paul Jurczak