TensorRT在ubuntu18.04的安裝
阿新 • • 發佈：2020-08-08
配置環境
ubuntu 18.04 ，cuda 10.1 ， cudnn 7.5 ， python3.5 ;
ubuntu 18.04 ，cuda 10.1 ， cudnn 7.5 ， python3.5 ;
安裝步驟
1.首先下載tar版本的安裝包，下載地址（ https://developer.nvidia.com/nvidia-tensorrt-5x-download）需要登陸NVIDIA。 安裝TensorRT前需要安裝Cuda和cudnn，安裝步驟可以參考 ubuntu安裝cuda和cudnn。 2. 開啟下載的TensorRT所在路徑，解壓下載的tar檔案：$ tar -xzvf TensorRT-XXX.tar.gz
3. 解壓新增環境變數
$ vim ~/.bashrc # 開啟環境變數檔案 # 將下面三個環境變數寫入環境變數檔案並儲存 export LD_LIBRARY_PATH=TensorRT解壓路徑/lib:$LD_LIBRARY_PATH export CUDA_INSTALL_DIR=/usr/local/cuda-9.0 export CUDNN_INSTALL_DIR=/usr/local/cuda-9.0 # 使剛剛修改的環境變數檔案生效 $ source ~/.bashrc
4. 下面是安裝Python的TensorRT包：進到解壓的TensorRT目錄下的Python目錄：
# 對於python2
$ sudo pip2 install tensorrt-XXX-cp27-cp27mu-linux_x86_64.whl
# 對於python3
$ sudo pip3 install tensorrt-XXX-cp35-cp35m-linux_x86_64.whl
5. 測試TensorRT是否安裝成功，進入Python編輯器載入tensorrt：
>>> import tensorrt >>> tensorrt.__version__
6. 錯誤分析： import tensorrt 時出現：
ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
解決方法：
cd /usr/local/cuda/lib64 sudo rm -rf libcudnn.so libcudnn.so.7 #刪除原有版本號，版本號在cudnn/lib64中查詢 sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成軟連線，注意自己下載的版本號 sudo ln -s libcudnn.so.7 libcudnn.so sudo ldconfig #立即生效