Win10环境用VS2019配置TensorRT 7.1.3.4

下面是我的环境

Win10
Anaconda3
Pyhton3.7
VS2019
cuda11.0
cudnn8.0

1、下载tensorrt

需要注册账号,后面用微信登陆很方便:官方链接
如果太慢的话,也可以下载我上传的:下载链接
下载后,解压,放到合适的位置

2、添加环境变量

此电脑->右键->属性
Win10环境用VS2019配置TensorRT 7.1.3.4
高级系统设置
Win10环境用VS2019配置TensorRT 7.1.3.4
环境变量
Win10环境用VS2019配置TensorRT 7.1.3.4

系统变量->Path(双击进入)
Win10环境用VS2019配置TensorRT 7.1.3.4
新建(下面是我的地址)
Win10环境用VS2019配置TensorRT 7.1.3.4

3、安装uff和graphsurgeon

cd D:\Software\TensorRT-7.1.3.4\graphsurgeon
pip install graphsurgeon-0.4.5-py2.py3-none-any.whl

Win10环境用VS2019配置TensorRT 7.1.3.4

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

Win10环境用VS2019配置TensorRT 7.1.3.4

4、复制tensorrt的dll到cuda安装目录

Win10环境用VS2019配置TensorRT 7.1.3.4
我的cuda安装目录是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin

5、测试

下载mnist数据集

 cd ../data/mnist
 python download_pgms.py

Win10环境用VS2019配置TensorRT 7.1.3.4
下载失败的话,可以下载我上传的资源:链接
解压放到 data/mnis t文件夹下面

用VS2019打开sample_mnist.sln
Win10环境用VS2019配置TensorRT 7.1.3.4
右键->属性
Win10环境用VS2019配置TensorRT 7.1.3.4

Win10环境用VS2019配置TensorRT 7.1.3.4
Win10环境用VS2019配置TensorRT 7.1.3.4

Win10环境用VS2019配置TensorRT 7.1.3.4
右键->生成
Win10环境用VS2019配置TensorRT 7.1.3.4

cd ../bin
sample_mnist.exe

Win10环境用VS2019配置TensorRT 7.1.3.4
参考博文:Tensorrt(I)win10安装配置,TensorRT,一,Win10

上一篇:C图形库==时钟


下一篇:LeetCode 240 - 搜索二维矩阵 II