Ubuntu Tensorrt安裝教程

2021-10-13 13:02:53 字數 2575 閱讀 7529

2、安裝tensorrt

version=

"7.x.x.x"

os="os"

arch=$(uname -m)

cuda=

"cuda-x.x"

cudnn=

"cudnn8.x"

tar xzvf tensorrt-$

.$.$

-gnu.$

.$.tar.gz

#7.***是您的tensorrt版本

#os是:ubuntu-16.04或ubuntu-18.04

#arch直接複製不需要改動

#cuda-xx 是cuda版本 10.2, 11.0, 要麼 11.1

#cudnn8.x 是cudnn版本 8.0

2、將絕對路徑新增到tensorrtlib 目錄到環境變數 ld_library_path

#使用以下命令開啟.bashrc檔案

sudo gedit ~

/.bashrc

#解壓得到tensorrt-7.2.1.6的資料夾,將裡邊的lib絕對路徑新增到環境變數中,使用者名稱和tensorrt的版本號更換為自己的

export ld_library_path=$ld_library_path:

/home/lig/tensorrt-

7.2.1

.6/lib

#儲存檔案並退出,執行以下命令,使路徑生效

source ~

/.bashrc

3、安裝python tensorrt wheel檔案

cp3x的x代表python的版本,如下圖,cp34代表python版本為3.4,這個需要根據自己電腦安裝的python版本進行選擇,選擇錯誤會報錯,大概意思是說tensorrt不能在這個平台上安裝

特別注意:

如果你想把tensorrt安裝到系統的python中,需要使用pip3;如果想裝到anconda的python環境中,直接使用pip,博主是安裝到anconda環境中,所以後續預設都使用pip

4、安裝python uff wheel檔案。只有當你將tensorrttensorflow一起使用時才需要安裝這個檔案

cd tensorrt-$

/uff

sudo pip install uff-

0.6.9

-py2.py3-none-

any.whl

5、安裝python graphsurgeon whl檔案

cd tensorrt-$

/graphsurgeon

sudo pip install graphsurgeon-

0.4.5

-py2.py3-none-

any.whl

6、安裝python onnx-graphsurgeon whl檔案

cd tensorrt-$

/onnx_graphsurgeon

sudo pip install onnx_graphsurgeon-

0.2.6

-py2.py3-none-

any.whl

如果你的電腦沒有安裝cmake,安裝這個檔案時會報找不到cmake這個模組的錯誤,使用以下命令安裝cmake,pip預設安裝到anconda,pip3預設安裝到系統自帶的python環境中。

7、驗證安裝

終端進入python環境,輸入import tensorrt,列印出tensorrt版本,即安裝成功

示例都在samples資料夾,裡面有c++和python示例

a、生成並執行其中乙個樣本,例如, 樣本samplemnist,示例執行方法都在readme.md檔案中

b、python示例位於 sampels/ python 目錄(提示沒有pycuda時,參考第8條)

8、安裝pycuda

pip install 'pycuda>=2019.1.1'

mongodb els memcached安全漏洞

elasticsearch漏洞 漏洞 cve 2014 3120 命令執行 cve 2015 3337 目錄穿越 cve 2014 3120 命令執行 啟動 docker compose build docker compose up d cve 2015 3337 目錄穿越 在安裝了具有 site...

pureftp mysql pureftp安裝部署

ftp 檔案傳輸 pure ftpd ftp軟體 作用 通過對映登入傳送和接收檔案。wget 2 解壓包 tar jxvf pure ftpd 1.0.43.tar.bz2 3 進入解壓目錄 cd pure ftpd.4 初始化安裝環境 先建立安裝目錄mkdir usr local pureftpd...

thinkphp nginx phpcgj安裝配置

環境 mysql 5.6.26 nginx 1.9.4.tar.gz php 5.6.13 程式框架thinkphp 客戶要求必須使用nginx php 1.首先安裝nginx mysql php 2.安裝擴充套件php擴充套件 pdo mysql 1 cd usr local src php 5....