python (12.9k questions)

javascript (9.2k questions)

reactjs (4.7k questions)

java (4.2k questions)

java (4.2k questions)

c# (3.5k questions)

c# (3.5k questions)

html (3.3k questions)

Questions - tensorrt

Cuda illegal memory access when running inference on *.engine

After exporting a YoloV5 model to .engine I receive an error when trying to perform inference on it. Loading model.engine for TensorRT inference... [01/16/2022-04:18:26] [TRT] [I] [MemUsageChange] Ini...
test-img

user17948924

tensorflow

pytorch

tensorrt

yolov5

Votes: 0

Answers: 1

Latest Answer

Yes this is a known YOLOv5 TRT issue with AutoShape. The following code works correctly: python export.py --weights yolov5s.pt --include engine python detect.py --weights yolov5s.engine But if we use...
test-img

Glenn Jocher

Posts

Questions

Blogs

Jobs