TensorRT在ubuntu18.04的安裝


配置環境
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 #立即生效
 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM