搭建 TensorFlow 的 GPU Docker 环境

TensorFlow 的 GPU 镜像

1
2

$ docker pull registry.cn-shanghai.aliyuncs.com/tcc-public/tensorflow:latest-cuda10.0-py3
$ nvidia-docker run --shm-size 8G --rm -v /exports:/exports -it registry.cn-shanghai.aliyuncs.com/tcc-public/tensorflow:latest-cuda10.0-py3 bash

在镜像中安装其它软件

（最好先更新软件源/root/.pip/pip.conf, /etc/apt/sources.list，以便更快下载）

进入 docker 内部

1	$ pip install pandas sklearn lightgbm catboost

说明

如需使用 GPU 的镜像，需要在宿主机安装 nvidia-docker 及 CUDA 相关工具，具体方法请见：

《搭建深度学习的 docker 环境》https://www.jianshu.com/p/cc9e617d79c2

搭建TensorFlow的GPUDocker环境

搭建 TensorFlow 的 GPU Docker 环境