当前位置: 首页 > 资讯

【原创】Docker镜像Deepo 一网打尽主流深度学习框架

论智       2017-11-02

Docker镜像Deepo打包了所有流行的深度学习框架。一方面,它为深度学习开发提供了方便,可以直接使用Deepo镜像,而不用自行搭建环境。另一方面,Deepo提供了一个统一的环境,为重现深度学习研究结果提供了便利。


Deepo Logo


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

cuda8.06.5-8.08.0

cudnnv5v2-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