yolov8 转wts文件（自定义数据集）失败

[pcycccccc]

Env

• GPU: RTX 3080. Env

• GPU: RTX 3080.

• OS, Win10.

• Cuda version 11.8

• TensorRT version 8.6.1

我在自己训练好的模型转wts的时候失败了，提示找不到模块 No module named 'ultralytics.utils'，这个pt模型是我用yolov8s.pt进行训练得到的，报错信息如下 但是，我试着转官方提供的yolov8n.pt时却能正确转换wts文件，不知道为啥会有这种问题....

自定义数据集的best.pt模型是我正常训练yolov8s得到的，之前搞其他C++部署时，转onnx模型是可以正常使用的。

补充一句：查过说是要pip install这个模块，但是我之前训练、验证、测试过程中是正常的，没有出现过提示找不到ultralytics模块，模快更新该模块也无法解决上述问题。

[pcycccccc]

另外，yolov8支持转engine模型文件，是否可以用这种方式来转换模型？并用在该项目提供的tensorRT代码来完成部署呢？

[wang-xinyu]

在https://github.com/ultralytics/ultralytics/tree/main 的主目录下执行gen_wts.py

[stale[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.