LINUX安裝TensorRT及特別注意事項

2021-08-28 13:43:19 字數 659 閱讀 7621

柳鯤鵬

這個在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.gz
cd python

sudo pip3 install tensorrt-4.0.1.6-cp35-cp35m-linux_x86_64.whl

重複:這裡的cp35,表示需要python3.5,高了低了好像都不行。

cd uff

sudo pip3 install uff-0.4.0-py2.py3-none-any.whl

這裡對python版本的要求就很寬鬆,python2/3都可以。

export ld_library_path=/usr/local/cuda/lib64:/home/gh/gh4ai/tensorrt-4.0.1.6/lib:$ld_library_path

source .bashrc
python3

import tensorrt

這樣就安裝成功!

tensorrt 安裝 新的

uname r 5.3.0 26 generic gpu1070 nvidia 430.50 cuda v10.0.130 cudnn 7604 tensorflow gpu 1.13.2 tensorrt 5.1.50 python 3.6.9 看 uff uff whl graphsurgeon...

TensorRT安裝及使用

如果出錯了看是不是這個問題 tensorrt官方安裝 公升級 解除安裝指南 下邊的內容我是學習b站上一位up主的教程,自己順便記錄一下學習的內容。import tensorrt as trt convert to uff frozen inference graph.pb model file da...

TensorRT安裝與使用(一)

參考安裝教程請直接轉到3.1 tar安裝 nvidia tensorrt 的核心是乙個c 庫,有助於對 nvidia 圖形處理單元 gpu 進行高效能推理。tensorrt 採用經過訓練的網路,該網路由網路定義和一組經過訓練的引數組成,並生成高度優化的執行時引擎,為該網路執行推理。tensorrt通...