我的环境:
- Anaconda3 / Python 3.7
- Tensorflow-gpu 1.14
- TensorRT 7
- CUDA 10
- CUDNN 7.6
- VS 2019
一、下载TensorRT
TensorRT 官方下载:地址
二、添加环境变量
PATH:D:\it\TensorRT-7.0.0.11\lib (这是我的)
三、安装uff和graphsurgeon
分别找到:graphsurgeon-0.4.1-py2.py3-none-any.whl 与 uff-0.6.5-py2.py3-none-any.whl
pip install graphsurgeon-0.4.1-py2.py3-none-any.whl
pip install uff-0.6.5-py2.py3-none-any.whl
四、复制dll文件到cuda安装目录
把上面 的 dll 拷贝到: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin (这是我的地址啦)
五、测试
VS2019 打开-》sample_mnist.sln
右键-》属性
右键-》生成
生成后文件的目录:
下载 mnist 数据:(python 运行即可)
运行 sample_mnist.exe