search

triple-Mu / YOLOv8-TensorRT

YOLOv8 using TensorRT accelerate !
MIT License
1.23k stars 210 forks source link

export engine error #198

Closed quxianjiuguo closed 1 month ago

quxianjiuguo commented 4 months ago

python build.py --weights yolov8n.onnx --iou-thres 0.65 --conf-thres 0.25 --topk 100 --fp16 --device cuda:0

I tried to run build.py to export engine. However, i encountered an error image

triple-Mu commented 4 months ago

python build.py --weights yolov8n.onnx --iou-thres 0.65 --conf-thres 0.25 --topk 100 --fp16 --device cuda:0

I tried to run build.py to export engine. However, i encountered an error image

It seems that you have installed an error pytorch. Try https://pytorch.org/get-started/locally/ to install pytorch.