E-Dreamer

脚踏实地,仰望星空

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理
配置环境
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 #立即生效
 
posted on 2020-08-08 09:33  E-Dreamer  阅读(2031)  评论(0编辑  收藏  举报