search

WongKinYiu / yolov9

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information
GNU General Public License v3.0
8.6k stars 1.31k forks source link

Export from pt to Tensorrt get error.. #484

Open blackCmd opened 1 month ago

blackCmd commented 1 month ago

I'm having trouble converting yolov9-e-converted.pt to a TensorRT model using export.py. I've tested this on Windows 10, 11, and Ubuntu 22.04, and I'm using cuda12.4.1 and tensorrt 10.0.1. I've encountered the following error:

image

How to solve this error 'tensorrt.tensorrt.IBuilderConfig' object has no attribute 'max_workspace_size'

Linaom1214 commented 1 month ago

I'm having trouble converting yolov9-e-converted.pt to a TensorRT model using export.py. I've tested this on Windows 10, 11, and Ubuntu 22.04, and I'm using cuda12.4.1 and tensorrt 10.0.1. I've encountered the following error:

image

How to solve this error 'tensorrt.tensorrt.IBuilderConfig' object has no attribute 'max_workspace_size'

you can try this method. https://github.com/Linaom1214/TensorRT-For-YOLO-Series

blackCmd commented 1 month ago

@Linaom1214 Geneius! It works!