我的環境:
- 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