如果出錯了看是不是這個問題:
tensorrt官方安裝、公升級、解除安裝指南:
下邊的內容我是學習b站上一位up主的教程,自己順便記錄一下學習的內容。
import tensorrt as trt
$convert-to-uff frozen_inference_graph.pb
>>>model_file = './data/mnist/mnsit.uff'
with builder = trt.builder(trt_logger) as builder,builder.create_network() as network,trt.uffparser() as parser:
parser.register_input('placeholder',(1,28,28))
parser.register_output('fc2/relu')
parser.parse(model_file,network)
TensorRT安裝與使用(一)
參考安裝教程請直接轉到3.1 tar安裝 nvidia tensorrt 的核心是乙個c 庫,有助於對 nvidia 圖形處理單元 gpu 進行高效能推理。tensorrt 採用經過訓練的網路,該網路由網路定義和一組經過訓練的引數組成,並生成高度優化的執行時引擎,為該網路執行推理。tensorrt通...
tensorrt安裝教程及失敗情況
tensorrt安裝教程 在進入 tensorrt samples資料夾執行make報錯 錯誤 dkpg query 沒有找到與cuda cudart 0 9 相匹配得軟體包 解決辦法 在 tensorrt samples 資料夾中有個 makefile.config 檔案,裡面第4行 cuda v...
LINUX安裝TensorRT及特別注意事項
柳鯤鵬 這個在tensorrt的python目錄下,可以看到包含有cpxx的whl檔案,就是對用的python版本號。解壓tar xvf tensorrt 4.0.1.6.ubuntu 16.04.4.x86 64 gnu.cuda 9.2.cudnn7.1.tar.gzcd python sudo...