概要
将训练好的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个变量.
<