tensorRT+windows
1、找到对应的cuda版本的tensorRT,下载后解压
2、在python环境中conda建立虚拟环境、安装tensorflow-gpu
3、pip安装tensorRT文件夹中graphsurgeon和uff
4、下载mnist数据
5、打开\TensorRT-6.0.1.5\data\mnist运行python generate_pgms.py生成pgm图像数据
6、打开data\samples\sampleMNIST里的sln工程,配置对应的sdk环境和vs环境(我的电脑是8.1和v140)
原创
2020-10-12 15:31:48 ·
1087 阅读 ·
0 评论