Ubuntu 16.04安装使用--Ubuntu16.04下基于Docker的Caffe-GPU版本环境搭建总结

本文介绍如何在Ubuntu环境下安装Docker,并通过Docker安装配置CUDA和Caffe,实现GPU支持,适用于深度学习环境搭建。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Ubuntu 16.04安装使用Docker

标签: ubuntulinuxdocker
34209人阅读 评论(0) 收藏 举报
分类:

目录(?)[+]

题记

安装系列一篇,最近闲暇来折腾折腾docker和spring boot。记录步骤,简要说明,目的是能够快速安装使用,方便查阅,这就是写安装文章的作用,我个人不喜欢写很多参数说明,有兴趣的人自会去深入阅读官方文档说明,就像app的理念一样,一次只做好一件事。


Linux环境

Ubuntu 16.04(LTS)

简单方法

$ sudo apt-get update
$ sudo apt-get install docker

搞定!
但是这种方法有一个缺点:安装的不一定是最新的docker,安装版本是1.10.2,不过,对于docker使用没任何什么问题,如果要后续升级方便,可以使用下面的方法。


另一种方法

1.切换到root权限或者用sudo

2.升级source列表并保证https和ca证书成功安装

# apt-get update
# apt-get install apt-transport-https ca-certificates

3.增加新的GPG 密钥

# apt-key adv --keyserver hkp://p80.pool.sks-keyservers.net:80 --recv-keys 58118E89F3A912897C070ADBF76221572C52609D

4.新增或编辑source列表里的docker.list文件

# vi /etc/apt/sources.list.d/docker.list  //如果不存在就新增

5.删除已有的entries

6.按照系统版本增加entry(Ubuntu Xenial 16.04 (LTS))

deb https://apt.dockerproject.org/repo ubuntu-xenial main

7.重新执行更新操作,并删除老的repo

# apt-get purge lxc-docker  //没有安装的话,跳过

8.查看是否有正确的可用版本

# apt-cache policy docker-engine

如图:

显示查找的版本

9.从14.04版本以上开始docker推荐安装linux-image-extra

# apt-get install linux-image-extra-$(uname -r)

10.安装docker

# apt-get update
# apt-get install docker-engine
# service docker start
# docker run hello-world

到此结束,the end!


page 155 of 366 in chapter 2016


---------------------------------------------------------  //   -------------------------------------------

下面的内容有点问题,需要注意。



blog.youkuaiyun.com/u011987514/article/details/70943378


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

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

  1. GNU/Linux x86_64 with kernel version > 3.10
  2. Docker >= 1.9 (official docker-enginedocker-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

  1. sudo apt-get -y install \  
  2.   apt-transport-https \  
  3.   ca-certificates \  
  4.   curl  
  5.   
  6. curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -  
  7.   
  8. sudo add-apt-repository \  
  9.        "deb [arch=amd64] https://download.docker.com/linux/ubuntu \  
  10.        $(lsb_release -cs) \  
  11.        stable"  
  12.   
  13. sudo apt-get update  

2.获取Docker CE

  1. sudo apt-get -y install docker-ce  

3.测试是否安装成功

  1. docker info  

4. 安装nvidia-docker


  1. # Install nvidia-docker and nvidia-docker-plugin  
  2. wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb  
  3. sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb  
官方提供的测试方法需要下载一个1G左右的镜像才能测试
这里只需要输入sudo nvidia_docker info测试一下即可
默认情况下Docker会把镜像安装在根目录下/var/lib/docker,这样镜像会大量占用系统盘空间,最终导致磁盘资源不足
解决方案是修改默认安装目录
由于我的/home磁盘资源比较多,所以都安装到/home去
  1. zcw@ubuntu:~# mkdir docker  
  2. zcw@ubuntu:~# vim /etc/default/docker  
添加配置信息
  1. DOCKER_OPTS="--graph=/home/docker"  
保存退出
  1. service docker restart  
发现配置并没有生效

解决方案:
  1. zcw@ubuntu:~# mkdir -p /etc/systemd/system/docker.service.d  
  2. zcw@ubuntu:~# cat /etc/systemd/system/docker.service.d/Using_Environment_File.conf  
如果没有该文件则自行创建,添加以下内容
  1. [Service]  
  2. EnvironmentFile=-/etc/default/docker  
  3. ExecStart=  
  4. ExecStart=/usr/bin/docker daemon -H fd:// $DOCKER_OPTS  

载入配置重启服务
  1. zcw@ubuntu:~# systemctl daemon-reload  
  2. zcw@ubuntu:~# service docker restart  
查看配置是否生效
  1. zcw@ubuntu:~# ps -ef|grep docker   


二、CUDA安装

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

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



三、Caffe

Caffe安装也很简单,大约2个G,只需等待即可
  1. 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 Engine Utility for NVIDIA GPUs

nvidia-gpu-docker

Documentation

The full documentation is available on the repository wiki.
A good place to start is to understand why nvidia-docker is needed in the first place.

Quick start

Assuming the NVIDIA drivers and Docker® Engine are properly installed (see installation)

Ubuntu distributions
# 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

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi
CentOS distributions
# 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.x86_64.rpm
sudo rpm -i /tmp/nvidia-docker*.rpm && rm /tmp/nvidia-docker*.rpm
sudo systemctl start nvidia-docker

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi
Other distributions
# 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_amd64.tar.xz
sudo tar --strip-components=1 -C /usr/bin -xvf /tmp/nvidia-docker*.tar.xz && rm /tmp/nvidia-docker*.tar.xz

# Run nvidia-docker-plugin
sudo -b nohup nvidia-docker-plugin > /tmp/nvidia-docker.log

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi
ppc64le (POWER) Archictecture

There is limited build support for ppc64le. Running make deb will build the nvidia-docker deb for ppc64le (if run on a ppc64le system). If the deb install fails because you have the 'docker.io' (>= v1.9) package installed, but not the 'docker-engine' package, you can force-install. There is currently no docker-provided docker-engine repository for ppc64le.

Not all the build targets for ppc64le have been implemented. If you would like for a Dockerfile to be created to enable a ppc64le target, please open an issue.

Issues and Contributing

A signed copy of the Contributor License Agreement needs to be provided to digits@nvidia.com before any change can be accepted.


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值