win10 + vs2019 + TensorRT 配置攻略

环境

  • Anaconda3 / Python 3.7
  • Tensorflow-gpu 2.0
  • TensorRT 7.3
  • CUDA 10.2
  • CUDNN 8.1
  • VS 2019

1.下载TensorRT

注意下载自己的对应版本

2.安装uff和graphsurgeon

分别找到:graphsurgeon-0.4.5-py2.py3-none-any.whl 与 uff-0.6.9-py2.py3-none-any.whl

>pip install uff-0.6.9-py2.py3-none-any.whl
>pip install graphsurgeon-0.4.5-py2.py3-none-any.whl 

3.复制dll文件到cuda安装目录

把上面 的 dll 拷贝到: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin (就是CUDA的安装路径)

4.添加环境变量

PATH:D:\xxxxxxx\TensorRT-7.2.3\lib

5.测试

VS2019 打开 -> sample_mnist.sln (懒得截了,借用别人的截图~~)

右键 -> 生成
生成后文件的目录:

注意先下载数据!

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


如何因为网络的原因下载不下来,可以把部分代码修改如下:

#改一下url的文件
with urllib.request.urlopen("http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz") as res:
     data = load_mnist_data(gzip.decompress(res.read()))

with urllib.request.urlopen("http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz") as res:
     labels = load_mnist_labels(gzip.decompress(res.read()))

运行 sample_mnist.exe

[参考链接]
https://blog.csdn.net/qq_19707521/article/details/104986296
https://blog.csdn.net/u011473714/article/details/95042856

推荐阅读更多精彩内容