From b91b15d3e763c3d668ca18ebab5a1d633a21ecfa Mon Sep 17 00:00:00 2001 From: Wang Xinyu Date: Fri, 27 May 2022 08:38:03 +0800 Subject: [PATCH] Update README.md --- yolov5/README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/yolov5/README.md b/yolov5/README.md index 64cf925..216c558 100644 --- a/yolov5/README.md +++ b/yolov5/README.md @@ -66,6 +66,9 @@ sudo ./yolov5 -d yolov5.engine ../samples // install python-tensorrt, pycuda, etc. // ensure the yolov5s.engine and libmyplugins.so have been built python yolov5_trt.py + +// Another version of python script, which is using CUDA Python instead of pycuda. +python yolov5_trt_cuda_python.py ``` # INT8 Quantization