Docker镜像Deepo打包了所有流行的深度学习框架。一方面,它为深度学习开发提供了方便,可以直接使用Deepo镜像,而不用自行搭建环境。另一方面,Deepo提供了一个统一的环境,为重现深度学习研究结果提供了便利。
Deepo包含以下深度学习框架:
theano
tensorflow
sonnet
pytorch
keras
lasagne
mxnet
cntk
chainer
caffe
torch
快速开始
首先安装Docker和nvidia-docker。
从Docker Hub拉取镜像:
docker pull ufoym/deepo
运行:
nvidia-docker run --rm ufoym/deepo nvidia-smi
交互式运行(退出后不会自动销毁):
nvidia-docker run -it ufoym/deepo bash
和主机共享配置和数据:
nvidia-docker run -it -v /host/data:/data -v /host/config:/config ufoym/deepo bash
目录的对应关系:
主机容器
/host/data/data
/host/config/config
查看包含的框架的版本(路径)
>>> import tensorflow
>>> print(tensorflow.__name__, tensorflow.__version__)
tensorflow 1.3.0
>>> import sonnet
>>> print(sonnet.__name__, sonnet.__path__)
sonnet ['/usr/local/lib/python3.5/dist-packages/sonnet']
>>> import torch
>>> print(torch.__name__, torch.__version__)
torch 0.2.0_3
>>> import keras
>>> print(keras.__name__, keras.__version__)
keras 2.0.8
>>> import mxnet
>>> print(mxnet.__name__, mxnet.__version__)
mxnet 0.11.0
>>> import cntk
>>> print(cntk.__name__, cntk.__version__)
cntk 2.2
>>> import chainer
>>> print(chainer.__name__, chainer.__version__)
chainer 3.0.0
>>> import theano
>>> print(theano.__name__, theano.__version__)
theano 0.10.0beta4+14.gb6e3768
>>> import lasagne
>>> print(lasagne.__name__, lasagne.__version__)
lasagne 0.2.dev1
>>> import caffe
>>> print(caffe.__name__, caffe.__version__)
caffe 1.0.0
; th
│ ______ __ | Torch7
│ /_ __/__ ________/ / | Scientific computing for Lua.
│ / / / _ \/ __/ __/ _ \ | Type ? for help
│ /_/ \___/_/ \__/_//_/ | https://github.com/torch
│ | http://torch.ch
│
│th>
与同类项目的比较
.modern-deep-learningdl-dockerjupyter-deeplearningDeepo
ubuntu16.0414.0414.0416.04
cuda❌8.06.5-8.08.0
cudnn❌v5v2-5v6
theano❌✔️✔️✔️
tensorflow✔️✔️✔️✔️
sonnet❌❌❌✔️
pytorch❌❌❌✔️
keras✔️✔️✔️✔️
lasagne❌✔️✔️✔️
mxnet❌❌❌✔️
cntk❌❌❌✔️
chainer❌❌❌✔️
caffe✔️✔️✔️✔️
torch❌✔️️✔️✔️
项目信息
许可 MIT
Docker Hub页面 https://hub.docker.com/r/ufoym/deepo/
机器人网原创文章,未经授权禁止转载。详情见转载须知
本文来自机器人网,如若转载,请注明出处:https://www.jqr.com/news/008384