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安装方法
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)