docker使用物理机gpu运行模型
背景:云物理机没安装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
时间会记录下一切。
背景:云物理机没安装tf相关环境,需要使用docker直接跑模型
docker run -v /data/bert:/app --runtime=nvidia -it tensorflow/tensorflow:1.13.2-gpu-py3 bash