赞
踩
第一:安装TensorRT(FasterTransformer 支持cuda10.0,所以TensorRT也下cuda10.0对应版本)
1、下载TensorRT
https://developer.nvidia.com/nvidia-tensorrt-5x-download
2、安装pycuda
pip install pycuda
3、安装 TensorRT
- #在home下新建文件夹,命名为tensorrt_tar,然后将下载的压缩文件拷贝进来解压
- tar xzvf TensorRT-5.0.2.6.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.3.tar
-
- #解压得到TensorRT-5.0.2.6的文件夹,将里边的lib绝对路径添加到环境变量中
- export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/lthpc/tensorrt_tar/TensorRT-5.0.2.6/lib
-
- #安装TensorRT
- cd TensorRT-5.0.2.6/python
- #if python2
- sudo pip2 install tensorrt-5.0.2.6-py2.py3-none-any.whl
- #if python3
- sudo pip3 install tensorrt-5.0.2.6-py2.py3-none-any.whl
-
- #安装UFF
- cd TensorRT-5.0.2.6/uff
- #if python2
- sudo pip2 install uff-0.5.5-py2.py3-none-any.whl
- #if python3
- sudo pip3 install uff-0.5.5-py2.py3-none-any.whl
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。