天天看點

ubuntu安裝TensorRT

ubuntu18.04+cuda11.2+cudnn8.0.5+Tensorrt7.2.2.3安裝

環境

系統:ubuntu 18.04

編譯器:gcc 7.5

如果之前安裝過deb格式的tensorrt,需要解除安裝

如果用pip安裝過對應的包(根據自己安裝過的包解除安裝)

由于之前安裝過其他版本的tensorrt,安裝新的tensorrt時會預設安裝key中最新版本的tensorrt,是以想要安裝老版本的tensorrt需要删除其他版本的key。

檢視key

檢視key清單

删除key

删除安裝包

官網教程

cuda和cudnn也要是deb安裝

驗證

進入<code>/var/nv-tensorrt-repo-ubuntu1804-cuda11.1-trt7.2.3.4-ga-20210226</code>

安裝剩下的包,eg:uff

如果要安裝python接口,下載下傳壓縮包版本的tensorrt,在解壓後的目錄中通過pip安裝對應的whl

<code>sudo pip3 install TensorRT-${version}/python/tensorrt-*-cp3x-none-linux_x86_64.whl</code>

可能會遇到部分包缺失依賴

安裝對應的依賴

如果提示資料不存在,進入<code>data/mnist</code>目錄,運作<code>download_pgms.py</code>下載下傳資料

解壓壓縮包

寫入路徑

如果要安裝python接口,在解壓後的目錄中通過pip安裝對應的whl