当我们需要配置某些容易出现兼容性问题的环境时,虚拟环境(如Docker)可以说是一个很好的选择。那么怎么使用Docker构建caffe-gpu环境呢?下面简单说一下构建的步骤。
安装nvidia-docker
因为我们要使用gpu,而原生的docker不能很好地解决这个问题,因此推荐使用nvidia-docker。
安装nvidia-docker很简单,只需要sudo apt install nvidia-docker
即可,并且用法和原版的docker几乎一模一样。
下载caffe官方镜像
通过命令docker pull bvlc/caffe:gpu
即可下载该镜像了,当然了,如果你有下载其它版本的caffe的需要,命令也是一样的,找到这个镜像,然后pull下来即可。
使用caffe-gpu镜像
通过该镜像构建一个容器的命令很简单,即nvidia-docker -ti -v path1:path2 bvlc/caffe:gpu /bin/bash
这里的-ti
即通过命令行的方式使用该容器,-v path1:path2
的作用是将主机的某路径path1
挂载到容器内的某路径path2
下,这样就能很方便地进行文件的处理和交换了。
接下来就进入了容器下的命令行模式了。
进入容器后一些可能需要做的事
在容器里,很多非必须的东西都是没有的,例如vi, vim, pip等等,如果我们需要这些工