tensorrt 安装 (新的)

1 相关设备和文件的参数

uname -r 5.3.0-26-generic
gpu 1070
nvidia 430.50
cuda V10.0.130
cudnn 7604
tensorflow-gpu 1.13.2
tensorrt 5.1.50
python 3.6.9

2 cuda和cudnn安装方法

看 :https://blog.csdn.net/weixin_39875161/article/details/89925021#%E6%9F%A5%E7%9C%8B%E7%9B%B8%E5%85%B3%E7%89%88%E6%9C%AC

3 tensorrt安装方法

uff/*uff*.whl

graphsurgeon/*graphsurgeon*.whl

python/tensorrt*cp36*.whl

用pip3安装这3个文件,就完了。

4 报错解决方法

import tensorrt as tr报错如下。恭喜你说明成功了,如下操作就可以用了。

ImportError: libnvinfer.so.5: cannot open shared object file: No such file o

(1)添加环境变量

export LD_LIBRARY_PATH=/home/cheng/TensorRT-5.1.5.0/lib:$LD_LIBRARY_PATH

感觉没啥用

(2)把TensorRT-5.1.5.0/lib下文件都拷贝到/usr/local/cuda/include/下面

 sudo cp /home/cheng/TensorRT-5.1.5.0/lib/*  /usr/local/cuda/include/

pycharm 报错

Traceback (most recent call last):
  File "/home/cheng/Documents/practice_work/practice_py/爬虫/test/bs4_test01.py", line 19, in <module>
    import tensorrt
  File "/usr/local/lib/python3.6/dist-packages/tensorrt/__init__.py", line 1, in <module>
    from .tensorrt import *
ImportError: libnvinfer.so.5: cannot open shared object file: No such file or directory

没解决,用jupyter代替pycharm吧.

cudnn与cuda与tensorflow对应版本

linux下:

windows下:

更多内容看这里:https://tensorflow.google.cn/install/source 

tensorflow-gpu==1.13 (from versions: 
0.12.1, 
1.0.0, 1.0.1, 1.1.0rc1, 1.1.0rc2, 1.1.0, 1.2.0rc0, 1.2.0rc1, 1.2.0rc2, 1.2.0, 1.2.1, 1.3.0rc0, 1.3.0rc1, 1.3.0rc2, 1.3.0, 1.4.0rc0, 1.4.0rc1, 1.4.0, 1.4.1, 1.5.0rc0, 1.5.0rc1, 1.5.0, 1.5.1, 1.6.0rc0, 1.6.0rc1, 1.6.0, 1.7.0rc0, 1.7.0rc1, 1.7.0, 1.7.1, 1.8.0rc0, 1.8.0rc1, 1.8.0, 1.9.0rc0, 1.9.0rc1, 1.9.0rc2, 1.9.0, 
1.10.0rc0, 1.10.0rc1, 1.10.0, 1.10.1, 1.11.0rc0, 1.11.0rc1, 1.11.0rc2, 1.11.0, 1.12.0rc0, 1.12.0rc1, 1.12.0rc2, 1.12.0, 1.12.2, 1.12.3, 1.13.0rc0, 1.13.0rc1, 1.13.0rc2, 1.13.1, 1.13.2, 1.14.0rc0, 1.14.0rc1, 1.14.0, 
2.0.0a0, 2.0.0b0, 2.0.0b1)

猜你喜欢

转载自blog.csdn.net/weixin_39875161/article/details/104086459