tensorrt 安裝 新的

2021-10-02 09:12:44 字數 2034 閱讀 4340

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/*graphsurgeon*.whl

python/tensorrt*cp36*.whl

用pip3安裝這3個檔案,就完了。

import tensorrt as tr報錯如下。恭喜你說明成功了,如下操作就可以用了。

importerror: libnvinfer.so.5: cannot open shared object file: no such file o

(1)新增環境變數

export ld_library_path=/home/cheng/tensorrt-5.1.5.0/lib:$ld_library_path
感覺沒啥用

(2)把tensorrt-5.1.5.0/lib下檔案都拷貝到/usr/local/cuda/include/下面

sudo cp /home/cheng/tensorrt-5.1.5.0/lib/*  /usr/local/cuda/include/
traceback (most recent call last):

file "/home/cheng/documents/practice_work/practice_py/爬蟲/test/bs4_test01.py", line 19, in import tensorrt

file "/usr/local/lib/python3.6/dist-packages/tensorrt/__init__.py", line 1, in from .tensorrt import *

importerror: libnvinfer.so.5: cannot open shared object file: no such file or directory

沒解決,用jupyter代替pycharm吧.

linux下:

windows下:

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通...

tensorrt安裝教程及失敗情況

tensorrt安裝教程 在進入 tensorrt samples資料夾執行make報錯 錯誤 dkpg query 沒有找到與cuda cudart 0 9 相匹配得軟體包 解決辦法 在 tensorrt samples 資料夾中有個 makefile.config 檔案,裡面第4行 cuda v...