背景：云物理机没安装tf相关环境，需要使用docker直接跑模型
- 在docker hub下载一个tensorflow gpu镜像
- 运行docker，直接进入bash，使用nvidia-smi正常看到现存，然后正常跑代码即可
docker run -v /data/bert:/app --runtime=nvidia -it tensorflow/tensorflow:1.13.2-gpu-py3 bash
相关文章：