windows 下安装 tensorRT
环境配置(头文件、静态库和动态库)
- 去这个地方下载对应的版本https://developer.nvidia.com/nvidia-tensorrt-8x-download
- 将 TensorRT-8.0.1.6\include中头文件 copy 到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include
- 将TensorRT-8.0.1.6\lib 中所有lib文件 copy 到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
- 将TensorRT-8.0.1.6\lib 中所有dll文件copy 到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
通过 visual studio 工具编译简单案例
- 用VS2019 打开 TensorRT-8.0.1.6\samples\sampleMNIST\sample_mnist.sln
- 右键工程-》重新生成
- 编译成功既可查看 …\TensorRT-8.0.1.6\bin 目录下的可执行文件
运行测试
-
第一种方法:直接visual studio编译后运行 -
第二种方法:进入TensorRT-8.0.1.6\bin,用cmd执行,sample_mnist.exe --datadir=d:\path\to\TensorRT-8.0.1.6\data\mnist\
|