Ubuntu16.04下Cuda10.2的Tensorrt7.0安裝及填坑指南

上一篇博客重新安裝顯卡驅動440和cuda10.2版本安裝好了cuda10.2,那麼這一篇講述如何安裝tensorrt7.0及遇到的坑。

安裝tensorrt7

先去官網下載ubuntu16.04,cuda10.2對應的tensorrt7版本,==建議先把deb版本和tar版本都下載下來存好,==如下圖
在這裏插入圖片描述
這裏注意的是如果cuda是用deb版本安裝的,那麼tensorrt就要使用deb版本安裝,否則會有可能報錯。

下載下來後根據官網提供的安裝指導安裝指導安裝即可。

第一步,運行以下command:

os="ubuntu1x04"
tag="cudax.x-trt7.x.x.x-ga-yyyymmdd"
sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub

sudo apt-get update
sudo apt-get install tensorrt cuda-nvrtc-10-2

對應的截圖如下:
在這裏插入圖片描述
在這裏插入圖片描述
第二步:
如果是Python2.7環境,運行:

sudo apt-get install python-libnvinfer-dev

將會安裝上python-libnvinfer
如果是Python3.x環境,運行:

sudo apt-get install python3-libnvinfer-dev

將會安裝上python3-libnvinfer
如果想使用TensorFlow的tensorrt,運行:

sudo apt-get install uff-converter-tf

將會安裝上graphsurgeon-tf

第三步,驗證是否安裝成功,運行:

dpkg -l | grep TensorRT

出現如下圖,表示安裝成功。
在這裏插入圖片描述

填坑1

然而,import tensorrt卻報錯了,提示ModuleNotFoundError: No module named ‘tensorrt’,如下圖:
在這裏插入圖片描述
這什麼鬼。。。。
解決辦法
解壓我們剛纔下載的tar安裝包,cd到xxxx/Tensorrt/Pythpon/路徑下,下面有幾個whl文件,pip安裝對應的whl文件即可,
再次import tensorrt就不會報錯了。

填坑2

import uff 報錯
在這裏插入圖片描述
解決辦法:
cd到xxxx/Tensorrt/uff/下,pip安裝即可。

以上一般就可以正常使用了,在/usr/local/下會生成一個tensorrt文件夾,裏面包含bin,data,samples四個文件夾,samples文件夾中是官方例程的源碼。

https://blog.csdn.net/zong596568821xp/article/details/86077553

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章