【已解决】Jetson Xavier NX 下 yolov8 tensorrt模型部署中RuntimeError: Expected all tensors...的问题解决

一、硬件：Jetson Xavier NX

二、任务：在Jetson Xavier NX上部署yolov8，并且使用tensorrt进行加速推理。

问题：在使用yolo指令（yolo export model=yolov8n.pt format=engine）将.pt文件转化为.engine文件时，发生报错：RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:

问题原因：pytorch版本不适配导致无法进行tensorrt加速。

错误版本：torch==1.12.0     torchvision==0.13.0

修改后版本：torch==1.14.0     torchvision==0.14.1

修改后：模型可以正常转换，且能够使用模型进行加速推理。

总结：本博客记录up在NX上部署yolov8，并使用tensorrt加速推理中遇到的一个问题，感谢各位看到最后。如果觉得对您有帮助请记得点赞收藏嗷！如果对在NX上部署yolov8相关步骤存在疑问，也欢迎大家在评论区留言交流！