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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章