环境:
- Ubuntu 18.04
- CUDA 10
- CUDNN 7.6.3
- TensorRT 7
一、 下载tar包
TensorRT官方:地址
二、 解压tar包
$ tar xzvf TensorRT-7.x.x.x.<os>.<arch>-gnu.cuda-x.x.cudnn7.x.tar.gz
-其中
6.x.x.x is your TensorRT version
<os> is:
Ubuntu-14.04
Ubuntu-16.04
Ubuntu-18.04
CentOS-7.6
<arch> is x86_64 or ppc64le
cuda-x.x is CUDA version 9.0, 10.0, or 10.1
cudnn7.x is cuDNN version 7.6
三、 添加环境变量
vi ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/catt/tensorRT_.70/TensorRT-7.0.0.11/lib
source ~/.bashrc
四、 安装TensorRT的python接口
cd TensorRT-7.x.x.x/python
pip install tensorrt-7.x.x.x-cp3x-none-linux_x86_64.whl
五、安装UFF(Tensorflow所使用的)
cd TensorRT-7.x.x.x/uff
pip install uff-0.7.5-py2.py3-none-any.whl
六、 安装graphsurgeon
cd TensorRT-7.x.x.x/graphsurgeon
pip install graphsurgeon-0.4.1-py2.py3-none-any.whl