win10 tensorrt安装

win10 tensorrt安装

下载地址：

https://developer.nvidia.com/nvidia-tensorrt-7x-download

需要注册，登录。

选择跟cuda对应的版本：

1. 添加环境变量，比如：

PATH：D:\it\TensorRT-7.1.3.4\lib   

2. 安装uff和graphsurgeon，下载demo 需要的data
cd D:\Software\TensorRT-7.1.3.4\graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl

cd ../uff
pip install uff-0.6.9-py2.py3-none-any.whl

下载mnist数据集
 cd ../data/mnist
 python download_pgms.py

3. 复制tensorrt的dll到cuda安装目录

也在tensorrt的lib目录下，比如：

D:\it\TensorRT-7.1.3.4\lib   

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/121860556