yolov5的wts文件转engine时遇到错误，如何解决？(操作系统-linux)

我是把从Windows上跑自己的训练集，用的yolov5s，只有一个类别；
然后把py文件转为wts文件拷贝到nano上；
这里报错是需要改yololayer。cu这个文件吗？