基于Docker构建caffe环境

前言：

之前在搭建caffe环境的时候，总是踩各种坑，为来不踩gpu的坑，只能委屈使用cpu版本的caffe，那么接下来的福利就是这一切都不需要你去做了，只需要下拉一个caffe官方的docker环境，即可使用。白嫖的感觉真好！

实现：

1.基础环境要求：

 2.安装nvidia-docker(v2.0):

参考：安装教程

3.更新docker下载源：

sudo chmod -R 777 /etc/docker
sudo cp /etc/docker/daemon.json /etc/docker/backup.json
sudo gedit /etc/docker/daemon.json

添加如下内容：

{
  "registry-mirrors": ["https://alzgoonw.mirror.aliyuncs.com"]    
}

重启docker：

systemctl restart docker
systemctl status docker

（重点）4.下拉一个caffe的官方镜像：

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

5.基于官方镜像创建一个caffe的容器进行使用： 

6.测试：

容器中运行caffe示例

1）下载测试示例

cd /opt/caffe
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

2）调用caffe（cpu）

./build/tools/caffe time -model=examples/mnist/lenet_train_test.prototxt

3)调用caffe（gpu）

./build/tools/caffe time --model=examples/mnist/lenet_train_test.prototxt -gpu 0

完成！