win10安裝TensorRT

AI浩 2021-09-18 13:52:10 阅读数:277

win10 win tensorrt

2.1 下載TensorRT

版本使用TensorRT 7.2.3 for Windows,CUDA版本是11.0,

下載地址:NVIDIA TensorRT 7.x Download | NVIDIA Developer

文件大小約500M,

2.2 配置環境變量

將下載好的文件賦值到C盤根目錄,然後將“C:\TensorRT-7.2.3.4\lib”添加環境變量。如下圖:

image-20210917195646407

2.3 安裝必要的whl文件

在shell裏面安裝下面三個whl文件。執行命令:

 pip install nvidia-pyindex
pip install graphsurgeon
pip install uff

2.4 複制dll文件到cuda安裝目錄

image-20210917200405422

複制到:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin,如下圖:

image-20210917200842440

到這裏就安裝完成了,下面我們測試一下。

2.5 測試

使用VS2017,打開C:\TensorRT-7.2.3.4\samples\sampleMNIST的sample_mnist.sln,如圖:

image-20210917202315574

然後右鍵sample_mnist項目,右鍵選擇屬性。然後再屬性頁選擇“VC++目錄——可執行文件目錄”,打開“可執行文件目錄”,將“C:\TensorRT-7.2.3.4\lib”添加進去,然後點擊“確定”。

image-20210917202824489

選擇“C/C++——附加包含目錄,點進去添加“C:\TensorRT-7.2.3.4\bin”,然後點擊“確定”。

image-20210917203339127

右鍵“sample_mnist”工程,然後選擇“重新生成”。

image-20210917203732002

看到如下信息,則錶明生成成功。

image-20210917203844681

生成後的文件如下圖:

image-20210917203954178

然後執行在data-》mnist下面運行“download_pgms.py”文件,下載數據集。

image-20210917204041426

image-20210917201757453

然後執行調試。如下圖

image-20210917204247905

出現下圖則錶明成功。

image-20210917202237826

版权声明:本文为[AI浩]所创,转载请带上原文链接,感谢。 https://gsmany.com/2021/09/20210918135210397l.html