linux下基于docker安装caffe并使用

安装docker2参考:https://blog.csdn.net/ly_twt/article/details/105765730

1、安装

https://hub.docker.com/search/?isAutomated=0&isOfficial=0&page=1&pullCount=0&q=caffe&starCount=0

上面链接中有各个版本的caffe docker,随便选择一个即可。我我选择的是bvlc/caffe。

docker pull bvlc/caffe:gpu

可能会出现以下错误:

原因:执行该命令需要root权限。

解决办法:在命令前加上sudo。

下载中:

下载完成:

2、测试

(1)方式一

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

出现如下图则说明安装成功:

(2)方式二

 sudo docker images

出现如下图则说明安装成功:

3、docker下使用caffe

(1)

 sudo docker run -t -i bvlc/caffe:gpu /bin/bash

出现下图说明我们进入了caffe容器中:

(2)

执行

ls /opt

会出现:

(3)测试例子

①进入到/opt/caffe/data/mnist文件夹中:

②执行./get_mnist.sh,获取mnist数据集:

③查看/opt/caffe/data/mnist下的文件:

④要是在服务器上,可以训练试试,我这暂时不允许,所以省略了。

继续训练的话:

# cd到/opt/caffe目录下执行如下命令。转换格式,在examples/mnist生成了两个目录:mnist_test_lmdb和mnist_train_lmdb
./examples/mnist/create_mnist.sh

然后:

# 在/opt/caffe文件夹下运行
./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

(4)根据个人情况使用caffe。

(5)容器中的数据保存

按Ctrl+D或者输入exit退出当前容器。

退出后,如果你想重新再回到之前的容器继续使用,可以通过以下命令重启之前的容器:

$ docker start container_ID
$ docker attach container_ID
# 例如:
# sudo docker start ad9f3c53fd15
# sudo docker attach ad9f3c53fd15

如果不记得容器ID,可以使用:docker ps -a查看已经停止的容器信息。
(6)实现主机和容器之间的数据交互,需要通过docker Volume来完成。

主机文件cp到容器内:

# docker cp 需要拷贝的文件或者目录   容器名称:容器目录
# 比如:
sudo docker cp /home/ly/test-file ad9f3c53fd15:/opt

容器内文件cp到主机内:

# docker cp 容器名称:容器目录    宿主机目录
# 比如:
sudo docker cp ad9f3c53fd15:/opt/test-file /home/ly

(7)docker删除指令

删除指定容器:

sudo docker rm container_ID/name

删除指定镜像:

sudo docker rmi image_name

*****************************************后续还会继续更新*********************************************

发布了75 篇原创文章 · 获赞 9 · 访问量 9563

猜你喜欢

转载自blog.csdn.net/ly_twt/article/details/105771924
今日推荐