安装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
*****************************************后续还会继续更新*********************************************