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

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

All in one docker

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

包含的框架及系統依賴

  • 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 version

docker pull floydhub/dl-docker:cpu

RUN

CPU Version

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