1、找到对应的cuda版本的tensorRT,下载后解压

将lib文件夹中的dll拷贝到安装cuda的lib文件夹中

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)

7、编译成功即可运行bin文件夹中的sample_mnist.exe,如果编译没有成功查看配置依赖项等

tensorRT+windows

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-10-22
  • 2021-04-17
  • 2021-05-05
  • 2021-07-09
  • 2022-02-02
猜你喜欢
  • 2022-12-23
  • 2022-01-12
  • 2021-07-29
  • 2021-10-18
  • 2021-11-24
  • 2022-12-23
相关资源
相似解决方案