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