推荐的基础镜像列表
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