docker-ubuntu18.4 部署trt

环境配置臻视意见恶心的事。不是臻视，是真是。

用的别人的dockerfile，因为用别的平台。具体语法就不说。

基础环境

ubuntu18.4; py3; gcc cuda10.1; cudnn7.6.5.

流程

COPY命令，然后run执行，我看了下cudnn和trt加起来不到2g，可以放在一起执行。

cuda 10.1

正常安装，环境变量会自己

cudnn 7.6.5

把cuda下东西 mv到/usr/local/cuda-10.1下面。然后加入环境变量。

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PTH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PTH

如果不加入环境变量会报错、

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

然后

ln -s libcudnn.so.7.6.5  libcudnn.so.7

不做软连接，会报下面的错

 ImportError: /usr/local/cuda/lib64/libcudnn.so.7: file too short

trt 

解压后，需要添加到环境变量

export LD_LIBRARY_PATH=/project/TensorRT-6.0.1.5/lib:$LD_LIBRARY_PATH

ImportError: libnvinfer.so.6: cannot open shared object file: No such file or directory

就是上面这一条没加入环境变量。

pip安装把 python/ uff/ graph.../这三个下面的whl文件。

最后测试下

python -c "import tensort"

没问题就ok了。

重启重启后，如果发现环境变量没了，那么要在dckerfile中写入ENV 加入变量。