【环境搭建】在ubuntu上安装TensorRT

安装显卡驱动、CUDA以及cuDNN

【环境搭建】 在Ubuntu16.04上安装nvidiaGPU驱动和cuda
【环境搭建】 测试gpu的bandwidth；p2p的bandwidth测试；以及使用DeepBench测试conv、矩阵乘法的运算能力

ubuntu上安装TensorRT

1. 下载安装包
   https://developer.nvidia.com/nvidia-tensorrt-download
   
   下载对应cuda版本的tar压缩包
   
2. 添加环境变量

export LD_LIBRARY_PATH="TensorRT解压绝对路径/lib":$LD_LIBRARY_PATH
export LIBRARY_PATH="TensorRT解压绝对路径/lib":$LIBRARY_PATH
export CPLUS_INCLUDE_PATH="TensorRT解压绝对路径/include":$CPLUS_INCLUDE_PATH
export C_INCLUDE_PATH="TensorRT解压绝对路径/include":$C_INCLUDE_PATH
export CUDA_INSTALL_DIR="/usr/local/cuda":$CUDA_INSTALL_DIR
export CUDNN_INSTALL_DIR="/usr/local/cuda":$CUDNN_INSTALL_DIR

3. pip安装

cd TensorRT解压绝对路径/python

pip install 对应python版本的.whl

cd TensorRT解压绝对路径/uff

pip install uff的.whl

4. 测试
   在python中执行：

import tensorrt
tensorrt.__version__
# 查看tensorrt版本

结语

如果您有修改意见或问题，欢迎留言或者通过邮箱和我联系。
如果我的文章对您有帮助，转载请注明出处。