TensorRT (一) Win10 安装配置

我的环境:

  • 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

 

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页