caffe 環境準備

Docker 方式安裝:

安裝模式

1. cpu 模式

docker run -ti bvlc/caffe:cpu caffe --version

2. gpu 模式

nvidia-docker run -ti bvlc/caffe:gpu caffe --version

其他說明

1. 修改默認用戶及掛載路徑

docker run --rm -u $(id -u):$(id -g) -v $(pwd):$(pwd) -w $(pwd) bvlc/caffe:cpu caffe train --solver=example_solver.prototxt

2. 交互模式

docker run -ti bvlc/caffe:cpu ipython   # bash or ipython instead of caffe.

dokcer 安裝方式項目地址: https://github.com/BVLC/caffe/tree/master/docker