Deepo:幾乎包含所有主流深度學習框架的Docker鏡像

原文:Deepo
作者:Ming
譯者:Teixeira10

【譯者注】在本文中,作者簡單介紹了Deepo的使用,給出了清晰的步驟和命令,也將各個深度學習框架進行了比較。
以下爲譯文:

Deepo是一個幾乎包含所有流行深度學習框架的Docker映像,擁有一個完整的可複製的深度學習研究環境。它涵蓋了當前最流行的深度學習框架:
theano,tensorflow,sonnet,pytorch,keras,lasagne,mxnet,cntk,chainer,caffe,torch

快速啓動

安裝

步驟1:安裝Dockernvidia-docker
步驟2:獲得Deepo鏡像。

你可以直接從Docker Hub下載鏡像,或者自己構建鏡像。

選項1:從Docker Hub獲得鏡像(建議)

docker pull ufoym/deepo

選項2:在本地構建Docker鏡像

git clone https://github.com/ufoym/deepo.git
cd deepo && docker build -t ufoym/deepo .

請注意,這可能需要幾個小時,因爲它從頭開始編譯了一些庫。

用法

現在你可以試試這個命令:

nvidia-docker run --rm ufoym/deepo nvidia-smi

這是能夠運行的,並且使Deepo能夠在docker容器內使用GPU。如果這個方法不起作用,可以搜索nvidia-docker GitHub上的問題——這上面已經有許多解決方案。將一個交互式shell放入一個容器,該容器不會在你退出之後自動刪除。

nvidia-docker run -it ufoym/deepo bash

如果你想要在本地(你的機器或VM)和Deepo容器之間進行共享數據和配置,請使用-v。

vidia-docker run -it -v /host/data:/data -v /host/config:/config ufoym/deepo bash

現在,請開始你的Deepo之旅吧!

tensorflow

$ python

>>> import tensorflow
>>> print(tensorflow.__name__, tensorflow.__version__)
tensorflow 1.3.0

sonnet

$ python

>>>>>> import sonnet
>>> print(sonnet.__name__, sonnet.__path__)
sonnet ['/usr/local/lib/python3.5/dist-packages/sonnet']

pytorch

$ python

>>> import torch
>>> print(torch.__name__, torch.__version__)
torch 0.2.0_3

keras

$ python

>>> import keras
>>> print(keras.__name__, keras.__version__)
keras 2.0.8

mxnet

$ python

>>> import mxnet
>>> print(mxnet.__name__, mxnet.__version__)
mxnet 0.11.0

cntk

$ python

>>> import cntk
>>> print(cntk.__name__, cntk.__version__)
cntk 2.2

chainer

$ python

>>> import chainer
>>> print(chainer.__name__, chainer.__version__)
chainer 3.0.0

theano

$ python

>>> import theano
>>> print(theano.__name__, theano.__version__)
theano 0.10.0beta4+14.gb6e3768

lasagne

$ python

>>> import lasagne
>>> print(lasagne.__name__, lasagne.__version__)
lasagne 0.2.dev1

caffe

$ python

>>> import caffe
>>> print(caffe.__name__, caffe.__version__)
caffe 1.0.0

$ caffe –version

caffe version 1.0.0

torch

$ th

 │  ______             __   |  Torch7
 │ /_  __/__  ________/ /   |  Scientific computing for Lua.
 │  / / / _ \/ __/ __/ _ \  |  Type ? for help
 │ /_/  \___/_/  \__/_//_/  |  https://github.com/torch
 │                          |  http://torch.ch
 │
 │th>

選擇比較

軟件許可

Deepo有MIT許可證

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章