Could not load dynamic library ‘libnvinfer.so.6'

Could not load dynamic library ‘libnvinfer.so.6’

首先说明：我的是tensorflow-gpu2.1.0
cuda是10.1
cudnn是7.6.4
ubuntu18.04

你可以先去/usr/local/cuda-10.1/lib64下查看是否有libnvinfer.so.6，应该是没有的。

那么原因就是因为没有安装TensorRT，可以通过以下命令来安装：

sudo apt-get install -y --no-install-recommends libnvinfer6=6.0.1-1+cuda10.1 \
    libnvinfer-dev=6.0.1-1+cuda10.1 \
    libnvinfer-plugin6=6.0.1-1+cuda10.1

另外说以下我做tensorflow一个月来的感觉。在windows下的参考资料相对linux要少很多，前期做一些简单的东西的话，在windows上是可以的，但是涉及到自定义op之类的时候，就比较头大了，因为官方的介绍都只有linux和moc系统下的。windows下应该也是能做出来的，但是作为小白的我还是做不出来，只能乖乖下了linux系统。