caffe在docker环境中使用GPU

1.Prerequisites

参考这里https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(version-2.0) 按照步骤来就行.

这里作为试验就直接从网上pull 了一个，如果你要自己定制，你需要去了解Docker 相关的知识。

nvidia-docker run -ti bvlc/caffe:gpu caffe --version #关于如何加速下载，网上有很多的 daocloud 的只要运行一个脚本就行

dl_test_1:latest 是我自己命名的 ,这里有一个端口映射，我将Container 里面的ssh 服务映射到了50005号端口，所以可以通过50005号端口访问Container。 -v 用于挂载主机的目录，这样容器和主机就可以进行文件交换。

首先进入caffe 目录。这个是在/opt/caffe

./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

在CPU模式下运行

root@72f016eb6fce:/opt/caffe# ./build/tools/caffe time -model=examples/mnist/lenet_train_test.prototxt

结果

在GPU模式下运行：

root@72f016eb6fce:/opt/caffe# ./build/tools/caffe time --model=examples/mnist/lenet_train_test.prototxt -gpu 0

结果。

从结果可以看出使用GPU时时间缩短了很多

来源：CSDN

作者：hengmuling

链接：https://blog.csdn.net/diligentboy88/article/details/79016634

标签