概要
将训练好的YOLOX模型部署在windows系统下的GPU上。这里我选用的tensorrt版本为7.2.3、cuda版本为10.2。现在也有更好的cuda版本和tensorrt版本,后续若有时间再进行其他版本的尝试。
cuda10.2安装配置
1、下载cuda10.2
打开官网(https://developer.nvidia.com/cuda-toolkit-archive),选择10.2下载。

2、运行安装包,路径默认,选择精简安装模式,后面一路默认就OK,安装成功关闭。
3、配置环境变量。
安装结束后,右键 我的电脑–>属性–>高级系统设置–>环境变量,系统变量中已经加入了cuda的两个路径。再在系统变量path中加入下面的4个变量.




本文详细介绍了如何在Windows系统上使用CUDA 10.2、CUDNN 8.1.0和TensorRT 7.2.3部署YOLOX模型。内容包括CUDA的安装配置、CUDNN的下载与配置、TensorRT的安装与测试,以及从ONNX模型转换为TensorRT模型的步骤。
最低0.47元/天 解锁文章
2232

被折叠的 条评论
为什么被折叠?



