Caffe容器中如何用GPU跑mnist数据集

1. 首先查看机器的nvidia设备号：

ls -la /dev | grep nvidia

一般返回数据如下：

crw-rw-rw-  1 root root    195, 255 Oct 25 19:37 nvidiactl

2.进入Caffe容器。命令如下：
docker run -t --name wxb-caffe-20170115 --device /dev/nvidiactl --device /dev/uvm --device /dev/nvidia0 wxb/caffe /bin/bash
其中，--name wxb-caffe-20170115表示将这次运行创建的caffe容器命名为wxb-caffe-20170115，以众多container中标识它，方便下次使用。
3.用CPU还是GPU，可以在caffe根目录下的/example/mnist/lenet_solver.prototxt中修改。