包含深度學習經常使用框架的Docker環境

相關的代碼都在Github上,請參見個人Github,https://github.com/lijingpeng/deep-learning-notes
敬請多多關注哈~~~git

All in one docker

若是你不想單獨安裝每一個深度學習組件,而且厭倦於安裝過程當中的各類依賴衝突等問題,那麼推薦你使用Docker來搭建深度學習工做環境。下面是一個能夠參考的 All in one docker 環境。幾乎包含了全部的流行的深度學習框架,而且分別有CPU版本和GPU版本,與虛擬機不一樣的是,Docker幾乎沒有性能損失,所以你能夠放心的使用它。須要注意的是,GPU版本的Docker只能在Linux系統上運行。github

包含的框架及系統依賴

  • 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

build

CPU versiondocker

docker pull floydhub/dl-docker:cpu

RUN

CPU Versionbash

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

GPU Version框架

nvidia-docker run -it -p 8888:8888 -p 6006:6006 -v /sharedfolder:/root/sharedfolder floydhub/dl-docker:gpu bash
相關文章
相關標籤/搜索