本人工作環境為:
windows7 vs2015 cuda10.2
cudnn版本請用7.6.5,如果是8.0,運行的時候會報錯:計算機中丟失cudnn_7.dll
配置TensorRT流程如下:
1.首先下載TensorRT7.0
2.解壓在D盤根目錄,將D:\TensorRT-7.0.0.11\lib添加到系統環境變量Path中。
3.將D:\TensorRT-7.0.0.11\lib下的dll文件拷貝到:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
4.vs2015打開D:\TensorRT-7.0.0.11\samples\sampleMNIST
屬性里面:平台工具集選擇vs2015
a)將D:\TensorRT-7.0.0.11\lib加入 VC++目錄–>可執行文件目錄
b)將D:\TensorRT-7.0.0.11\include
和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include加入VC++目錄-->包含目錄
c)將D:\TensorRT-7.0.0.11\lib
和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64加入 VC++目錄–>庫目錄
d)將C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64\*.lib
myelin64_1.lib
nvinfer.lib
nvinfer_plugin.lib
nvonnxparser.lib
nvparsers.lib
加入鏈接器–>輸入–>附加依賴項
5.編譯
6.cmd進入D:\TensorRT-7.0.0.11\data\mnist,運行
python download_pgms.py會下載一些文件
7.下載完成后執行,得到: