1. 程式人生 > >Docker 快速搭建Caffe平臺

Docker 快速搭建Caffe平臺

(建議在翻牆的情況下進行,不然更新源的時候會出現無法下載的問題)

點選下載所需檔案
https://github.com/BVLC/caffe/tree/master/docker
1.使用docker下載Caffe映象

docker build -t caffe:cpu standalone/cpu

2.測試docker映象是有安裝成功

docker run -ti caffe:cpu caffe --version

如果出現以下資訊證明安裝執行成功。

libdc1394 error: Failed to initialize libdc1394
caffe version
1.0.0-rc3

3.用映象執行一個Caffe例項:

docker run -ti caffe:cpu bash -c "cd /opt/caffe/build; make runtest"

儲存映象的輸出到本地/workspace下面

docker run -ti --volume=$(pwd):/workspace caffe:cpu caffe train --solver=example_solver.prototxt

給予當前使用者root許可權

docker run -ti --volume=$(pwd):/workspace -u $(id -u):$(id -g) caffe:
cpu caffe train --solver=example_solver.prototxt

使用python-Shell

docker run -ti caffe:cpu python