下面是我的环境
Win10
Anaconda3
Pyhton3.7
VS2019
cuda11.0
cudnn8.0
1、下载tensorrt
需要注册账号,后面用微信登陆很方便:官方链接
如果太慢的话,也可以下载我上传的:下载链接
下载后,解压,放到合适的位置
2、添加环境变量
此电脑->右键->属性
高级系统设置
环境变量
系统变量->Path(双击进入)
新建(下面是我的地址)
3、安装uff和graphsurgeon
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
4、复制tensorrt的dll到cuda安装目录
我的cuda安装目录是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
5、测试
下载mnist数据集
cd ../data/mnist
python download_pgms.py
下载失败的话,可以下载我上传的资源:链接
解压放到 data/mnis t文件夹下面
用VS2019打开sample_mnist.sln
右键->属性
右键->生成
cd ../bin
sample_mnist.exe
参考博文:Tensorrt(I)win10安装配置,TensorRT,一,Win10