search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.28k stars 198 forks source link

yolov7 的代码中运行过程中出现Error Code 1: Cuda Runtime (an illegal memory access was encountered) #24

Closed xuexisuanfa closed 1 year ago

xuexisuanfa commented 1 year ago

yolov7 的代码中运行过程中出现Error Code 1: Cuda Runtime (an illegal memory access was encountered)， 想咨询一下作者您用所用pt 文件是哪个然后怎么转成trt文件？

FeiYull commented 1 year ago

@xuexisuanfa git clone https://github.com/WongKinYiu/yolov7

“git checkout” is necessary

git checkout 072f76c72c641c7a1ee482e39f604f6f8ef7ee92

640

python export.py --weights yolov7-tiny.pt --dynamic --grid python export.py --weights yolov7.pt --dynamic --grid python export.py --weights yolov7x.pt --dynamic --grid

1280

python export.py --weights yolov7-w6.pt --dynamic --grid --img-size 1280

xuexisuanfa commented 1 year ago

@xuexisuanfa git clone https://github.com/WongKinYiu/yolov7

“git checkout” is necessary

git checkout 072f76c72c641c7a1ee482e39f604f6f8ef7ee92

640

python export.py --weights yolov7-tiny.pt --dynamic --grid python export.py --weights yolov7.pt --dynamic --grid python export.py --weights yolov7x.pt --dynamic --grid

1280

python export.py --weights yolov7-w6.pt --dynamic --grid --img-size 1280 你好您这个是转成onnx 的，但是你的代码好像使用trtmodel

FeiYull commented 1 year ago

@xuexisuanfa

python export.py --weights yolov7-tiny.pt --dynamic --grid

../../../../TensorRT-8.4.2.4/bin/trtexec --onnx=yolov7-tiny.onnx --saveEngine=yolov7-tiny.trt --buildOnly --minShapes=images:1x3x640x640 --optShapes=images:4x3x640x640 --maxShapes=images:8x3x640x640