原文: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/deepo1
2
选项2:在本地构建Docker镜像
git clone https://github.com/ufoym/deepo.git
cd deepo && docker build -t ufoym/deepo .1
2
3
请注意,这可能需要几个小时,因为它从头开始编译了一些库。
用法
现在你可以试试这个命令:
nvidia-docker run --rm ufoym/deepo nvidia-smi1
2
这是能够运行的,并且使Deepo能够在docker容器内使用GPU。如果这个方法不起作用,可以搜索nvidia