原文:Deepo
作者:Ming
譯者:Teixeira10
【譯者注】在本文中,作者簡單介紹了Deepo的使用,給出了清晰的步驟和命令,也將各個深度學習框架進行了比較。
以下爲譯文:
Deepo是一個幾乎包含所有流行深度學習框架的Docker映像,擁有一個完整的可複製的深度學習研究環境。它涵蓋了當前最流行的深度學習框架:
theano,tensorflow,sonnet,pytorch,keras,lasagne,mxnet,cntk,chainer,caffe,torch。
快速啓動
安裝
步驟1:安裝Docker和nvidia-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許可證。