tensorrtx报错：ModuleNotFoundError: No module named ‘utils.torch_utils‘，将pt模型转化为wts模型，需要在yolo工程下进行

在使用tensorrtx的时候报错：ModuleNotFoundError: No module named ‘utils.torch_utils’
在使用tensorrtx的时候，要进行egine加速，需要将pt模型转化为wts模型，需要在yolo工程下进行
在ultralytics/yolov5.git这个工程下有utils这个文件夹
Traceback (most recent call last):
File “gen_wts.py”, line 6, in
from utils.torch_utils import select_device
ModuleNotFoundError: No module named ‘utils.torch_utils’

 https://github.com/ultralytics/yolov5.git