本文档详细介绍了如何在Ubuntu 16.04上使用nvidia-docker构建caffe-gpu环境。首先，安装nvidia-docker以支持GPU，接着下载caffe的官方镜像。通过镜像创建容器，并挂载本地目录进行文件交互。进入容器后，可能需要安装额外的工具和更新源。测试caffe-gpu环境包括验证GPU支持、安装cudnn以及编译和配置caffe。最后，学习如何退出和重新进入容器。
当我们需要配置某些容易出现兼容性问题的环境时，虚拟环境（如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等等，如果我们需要这些工
最低0.47元/天 解锁文章
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
