今天主要想介绍一种特别快的配置caffe以及faster rcnn的方法,直接利用docker来安装caffe镜像。
1. 前言
Docker 是一个开源的应用容器引擎,让开发者可以打包他们的应用以及依赖包到一个可移植的容器中,然后发布到任何流行的 Linux 机器上。因此,docker给应用提供了一个从开发到上线均一致的代码环境,让代码的流水线变得简单不少。caffe是我们在做关于深度学习的相关工作中常用的一个框架,faster rcnn是目标检测中非常有效的一种方法,在这里我会介绍如何利用docker快速安装caffe以及faster rcnn。
2. 安装docker
sudo apt-get install docker.io
3. 获取 Caffe 的 Docker 镜像
先搜索caffe的相关镜像
sudo docker search caffe
它会将docker hub 上面已经上传的caffe相关镜像显示出来,我们一般会选择stars数较多的文件进行下载。
sudo docker pull tleyden5iwx/caffe-gpu-master
docker pull就是用来下载镜像的语句
4. 利用已经下下来的镜像文件创建一个新的容器
sudo nvidia-docker run -ti 镜像名
这时候成功创建的容器会有一串特殊的ID,比如6952cc9673a6
并且你会进入到容器里面,如下图所示