Caffe的GPU环境搭建在docker支持下并不困难，但是过程比较杂

所需依赖如下，本文会说明安装方法，帮助大家少走弯路

  1. GNU/Linux x86_64 with kernel version > 3.10
  2. Docker >= 1.9 (official docker-engine, docker-ce or docker-ee only)
  3. NVIDIA GPU with Architecture > Fermi (2.1)
  4. NVIDIA drivers >= 340.29 with binary nvidia-modprobe

本文分成三部分总结

1.Docker安装与配置

2.Cuda安装

3.Caffe安装



一、Docker安装与配置

1.设置repository

sudo apt-get -y install \
  apt-transport-https \
  ca-certificates \
  curl

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo add-apt-repository \
       "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
       $(lsb_release -cs) \
       stable"

sudo apt-get update

2.获取Docker CE

sudo apt-get -y install docker-ce

3.测试是否安装成功

docker info

4. 安装nvidia-docker


# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb
官方提供的测试方法需要下载一个1G左右的镜像才能测试 
 

 


  这里只需要输入sudo nvidia_docker info测试一下即可 

 

 

 

 
 默认情况下Docker会把镜像安装在根目录下/var/lib/docker，这样镜像会大量占用系统盘空间，最终导致磁盘资源不足 

 

 
 解决方案是修改默认安装目录 

 

 
 由于我的/home磁盘资源比较多，所以都安装到/home去 

 

 
 
zcw@ubuntu:~# mkdir docker
zcw@ubuntu:~# vim /etc/default/docker
 

 

 
 添加配置信息 

 

 
  
 
DOCKER_OPTS="--graph=/home/docker"
 

 

 
 保存退出 

 

 
 
service docker restart
 

 

 
 发现配置并没有生效 

 

 
 
  

 

 
 解决方案： 

 

 
  
 
 
  
zcw@ubuntu:~# mkdir -p /etc/systemd/system/docker.service.d
zcw@ubuntu:~# cat /etc/systemd/system/docker.service.d/Using_Environment_File.conf
 
 
 

 

 
 如果没有该文件则自行创建，添加以下内容 

 

 
 
 
  
[Service]
EnvironmentFile=-/etc/default/docker
ExecStart=
ExecStart=/usr/bin/docker daemon -H fd:// $DOCKER_OPTS
 
  
 
 
 
 

   载入配置重启服务 
 
 
 
 
  
 
   
zcw@ubuntu:~# systemctl daemon-reload
zcw@ubuntu:~# service docker restart
 
  
 
 
 
 

   查看配置是否生效 
 
 
 
 
  
zcw@ubuntu:~# ps -ef|grep docker 
 
  
 
   
 
  
 
  
 
   
 
  
 
  
二、CUDA安装
 
 
 
 

   本部分完成第四项环境依赖： 
  NVIDIA drivers 
   >= 340.29 with binary  
  nvidia-modprobe 
 
 
 
 
  
 
 
 
 

   进入正题，到以下链接下载对应的CUDA,按官方教程，无脑完成 
 
 
  
 
 
  
 
 
 
 
 
   
  
 
 
 
 
 
  
 
   
 
  
 
  
 
  
 
  
三、Caffe
 
  

    Caffe安装也很简单，大约2个G,只需等待即可 
  
 
  
nvidia-docker run -ti bvlc/caffe:gpu caffe --version
 
  
 
 
 
 
 
  至此安装过程已经完成 
 
 
 
 
  
  
 
 
 
 
  默认情况下caffe以root身份运行,因此任何输出文件为root拥有，你可以通过flags来修改以及挂载目录 
 
 
 
 
   
  
docker run --rm -u $(id -u):$(id -g) -v $(pwd):$(pwd) -w $(pwd) bvlc/caffe:gpu caffe train --solver=example_solver.prototxt
 
  
 
  
 
 
 
 
 
  
  
 
 
 
 
  
 
 
 
 
 

 
  

        # docker
      
        # caffe
      
        # ubuntu
      
 
 GitHub 加速计划 / li / linux-dash
 
10.39 K
            
 
1.2 K
            
 
下载
            
 
 

          A beautiful web dashboard for Linux
        
 

          最近提交(Master分支：2 个月前 )
 
186a802e 
added ecosystem file for PM2 4 年前 
5def40a3 
Add host customization support for the NodeJS version 4 年前 
 
 
 Logo 
GitCode 开源社区
 
旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。
 
 
  
更多推荐
  
 
[转载]在Windows环境下安装GNU Radio
 
转自：在Windows环境下安装GNURadio_恐弱智_新浪博客GNU Radio是用Python开发的，大部分开源的工程能够在Linux环境下运行良好，而Windows下却运行的很勉强，而且安装配置都很复杂。GNU Radio算是个例外了，不光提供了Windows的二进制安装，还有比较详细的说明。我是Python小白，所以折腾了好久才弄好，特意记录下来，免得以后再装还折腾。GNU Radio的
 
avatar GitCode 开源社区
 
 
centOS 8 使用dnf安装Docker
 
DNF是什么？CentOS 8使用YUM软件包管理器版本v4.0.4。现在，该版本使用DNF(已删除YUM)。DNF是软件包管理器。它会在Linux发行版上安装，执行更新并删除软件包。使用DNF安装Docker跳过具有损坏依赖性的程序包一个有效的解决方案是使您的CentOS 8系统使用以下--nobest命令安装最符合条件的版本：sudo dnf install docker...
 
avatar GitCode 开源社区
 
 
定时同步数据库表(mysql+linux+crontab)
 
sync.sh里面的参数需要改变，ip/username/password/database/tablesync.sh#!/bin/sh# Please change the IP and password of the data source db.# Then change the table name.filename=/home/nington/db/$(date +%Y-%m
 
avatar GitCode 开源社区