搭建TensorFlow的GPU Docker环境

推荐的基础镜像列表

https://tianchi.aliyun.com/forum/postDetail?postId=67720

TensorFlow的GPU镜像

$ 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内部

$ pip install pandas sklearn lightgbm catboost

说明

如需使用GPU的镜像，需要在宿主机安装nvidia-docker及CUDA相关工具，具体方法请见：
《搭建深度学习的docker环境》https://www.jianshu.com/p/cc9e617d79c2