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

TensorRT (一) Win10 安装配置_第1张图片

分别找到: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安装目录

TensorRT (一) Win10 安装配置_第2张图片

把上面 的 dll 拷贝到: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin     (这是我的地址啦)

 

五、测试

VS2019 打开-》sample_mnist.sln

TensorRT (一) Win10 安装配置_第3张图片

 

右键-》属性

TensorRT (一) Win10 安装配置_第4张图片

TensorRT (一) Win10 安装配置_第5张图片

TensorRT (一) Win10 安装配置_第6张图片

 

右键-》生成

TensorRT (一) Win10 安装配置_第7张图片

 

生成后文件的目录:

TensorRT (一) Win10 安装配置_第8张图片

 

下载 mnist 数据:(python 运行即可)

TensorRT (一) Win10 安装配置_第9张图片

 

运行 sample_mnist.exe

TensorRT (一) Win10 安装配置_第10张图片

 

你可能感兴趣的:(TensorRT,机器学习,python)