当前位置:   article > 正文

FasterTransformer 安装

fastertransformer

第一:安装TensorRT(FasterTransformer 支持cuda10.0,所以TensorRT也下cuda10.0对应版本)

1、下载TensorRT

https://developer.nvidia.com/nvidia-tensorrt-5x-download

2、安装pycuda

pip install pycuda

3、安装 TensorRT

  1. #在home下新建文件夹,命名为tensorrt_tar,然后将下载的压缩文件拷贝进来解压
  2. tar xzvf TensorRT-5.0.2.6.Ubuntu-16.04.4.x86_64-gnu.cuda-9.0.cudnn7.3.tar
  3. #解压得到TensorRT-5.0.2.6的文件夹,将里边的lib绝对路径添加到环境变量中
  4. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/lthpc/tensorrt_tar/TensorRT-5.0.2.6/lib
  5. #安装TensorRT
  6. cd TensorRT-5.0.2.6/python
  7. #if python2
  8. sudo pip2 install tensorrt-5.0.2.6-py2.py3-none-any.whl
  9. #if python3
  10. sudo pip3 install tensorrt-5.0.2.6-py2.py3-none-any.whl
  11. #安装UFF
  12. cd TensorRT-5.0.2.6/uff
  13. #if python2
  14. sudo pip2 install uff-0.5.5-py2.py3-none-any.whl
  15. #if python3
  16. sudo pip3 install uff-0.5.5-py2.py3-none-any.whl
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/357670
推荐阅读
相关标签
  

闽ICP备14008679号