- 博客(2)
- 收藏
- 关注
原创 TensorRT加速推理(基于yolo5)
E:\yolov5-master\yolov5-master\yolov5s.pt (yolov5s.pt路径)E:\yolov5-master\yolov5-master\export.py (export.py路径)(E:\TensorRT\TensorRT-8.5.3.1\bin\trtexec)打开cmd终端命令:输出。在yolov5环境变量的终端输出:(直接python终端也可)--橙色字为自己的文件路径。(必须在自己的TensorRT的trtexec.exe文件夹下)生成engine加速文件。
2023-12-30 12:19:18
711
原创 安装好CUDA后CUDNN+TensorRT部署
3 将cuda\include中的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include。2 将cuda\bin中的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin。4 将cuda\lib中的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib。
2023-12-30 09:34:03
450
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人