Docker 方式安装:
安装模式
- cpu 模式
docker run -ti bvlc/caffe:cpu caffe --version
- gpu 模式
nvidia-docker run -ti bvlc/caffe:gpu caffe --version
其他说明
- 修改默认用户及挂载路径
docker run --rm -u $(id -u):$(id -g) -v $(pwd):$(pwd) -w $(pwd) bvlc/caffe:cpu caffe train --solver=example_solver.prototxt
- 交互模式
docker run -ti bvlc/caffe:cpu ipython # bash or ipython instead of caffe.
dokcer 安装方式项目地址: https://github.com/BVLC/caffe/tree/master/docker