【环境搭建】在ubuntu上安装TensorRT
安装显卡驱动、CUDA以及cuDNN
【环境搭建】 在Ubuntu16.04上安装nvidiaGPU驱动和cuda
【环境搭建】 测试gpu的bandwidth;p2p的bandwidth测试;以及使用DeepBench测试conv、矩阵乘法的运算能力
ubuntu上安装TensorRT
- 下载安装包
https://developer.nvidia.com/nvidia-tensorrt-download
下载对应cuda版本的tar压缩包
- 添加环境变量
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
- pip安装
cd TensorRT解压绝对路径/python
pip install 对应python版本的.whl
cd TensorRT解压绝对路径/uff
pip install uff的.whl
- 测试
在python中执行:
import tensorrt
tensorrt.__version__
# 查看tensorrt版本
结语
如果您有修改意见或问题,欢迎留言或者通过邮箱和我联系。
如果我的文章对您有帮助,转载请注明出处。