Docker 快速搭建Caffe平台

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_30115765/article/details/52995320

(建议在翻墙的情况下进行,不然更新源的时候会出现无法下载的问题)

点击下载所需文件
https://github.com/BVLC/caffe/tree/master/docker
1.使用docker下载Caffe镜像

docker build -t caffe:cpu standalone/cpu

2.测试docker镜像是有安装成功

docker run -ti caffe:cpu caffe --version

如果出现以下信息证明安装运行成功。

libdc1394 error: Failed to initialize libdc1394
caffe version 1.0.0-rc3

3.用镜像运行一个Caffe实例:

docker run -ti caffe:cpu bash -c "cd /opt/caffe/build; make runtest"

保存镜像的输出到本地/workspace下面

docker run -ti --volume=$(pwd):/workspace caffe:cpu caffe train --solver=example_solver.prototxt

给予当前用户root权限

docker run -ti --volume=$(pwd):/workspace -u $(id -u):$(id -g) caffe:cpu caffe train --solver=example_solver.prototxt

使用python-Shell

docker run -ti caffe:cpu python

猜你喜欢

转载自blog.csdn.net/qq_30115765/article/details/52995320
今日推荐