Windows10下Docker搭建深度学习框架

发表于   |   分类于

概述

Deepo

Deepo是一个几乎包含所有流行深度学习框架的Docker映像，拥有一个完整的可复制的深度学习研究环境。它涵盖了

  • theano
  • tensorflow
  • sonnet
  • pytorch
  • keras
  • lasagne
  • mxnet
  • cntk
  • chainer
  • caffe
  • torch

安装

直接从Docker Hub下载Deepo镜像。Github地址

  • CPU版本
docker pull ufoym/deepo:cpu
  • GPU版本
docker pull ufoym/deepo

运行

打开powershell执行以下命令，以bash方式运行该容器。

docker run -it ufoym/deepo:cpu bash

DL-docker

包含的框架及系统依赖

  • Ubuntu 14.04
  • CUDA 7.5 (GPU version only)
  • cuDNN v4 (GPU version only)
  • Tensorflow
  • Caffe
  • Theano
  • Keras
  • Lasagne
  • Torch (includes nn, cutorch, cunn and cuDNN bindings)
  • iPython/Jupyter Notebook (including iTorch kernel)
  • Numpy, SciPy, Pandas, Scikit Learn, Matplotlib
  • A few common libraries used for deep learning

安装

  • cpu版本
docker pull floydhub/dl-docker:cpu

运行

docker run -it -p 8888:8888 -p 6006:6006 -v /sharedfolder:/root/sharedfolder floydhub/dl-docker:cpu bash

运行以上命令后，会连接到bash，执行jupyter notebook。

mark

然后在Windows的浏览器中输入localhost:8888，会连接到该容器的Jupyter notebook

mark