无法将YOLOv8-Pose模型转换为TensorRT的问题求助

错误提示明确要求TensorRT模型转换必须在GPU环境下执行，即便你的系统已配置CUDA，默认情况下模型仍可能加载到CPU上，只需在代码中指定GPU设备即可解决问题。

修改后的代码（两种方式任选其一） #

方式1：加载模型时指定GPU设备

from ultralytics import YOLO
# 加载模型并指定使用第1块GPU（device=0）
model = YOLO('yolov8n-pose.pt', device=0)
# 导出为TensorRT格式
model.export(format='engine')

方式2：导出时指定GPU设备

from ultralytics import YOLO
model = YOLO('yolov8n-pose.pt')
# 导出时指定使用第1块GPU
model.export(format='engine', device=0)

额外说明 #

从你的错误日志可以看到，PyTorch版本为torch-2.0.1+cu118，已正确支持CUDA，只需确保GPU驱动正常、执行代码时GPU资源未被占用即可顺利完成转换。

内容的提问来源于stack exchange，提问作者Syed Muhammad Musa Kazmi