ubuntu16.04下通过nvidia-docker构建caffe-gpu环境

当我们需要配置某些容易出现兼容性问题的环境时,虚拟环境(如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
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值