docker使用物理机gpu运行模型

背景：云物理机没安装tf相关环境，需要使用docker直接跑模型

1. 在docker hub下载一个tensorflow gpu镜像
2. 运行docker，直接进入bash，使用nvidia-smi正常看到现存，然后正常跑代码即可

docker run -v /data/bert:/app --runtime=nvidia -it tensorflow/tensorflow:1.13.2-gpu-py3 bash