ITPub博客

首页 > 云计算 > > Ubuntu16.04安装nvidia-docker

Ubuntu16.04安装nvidia-docker

作者:浪花小雨滴 时间:2020-05-11 18:22:22 0 删除 编辑

最近部门打算买一波服务器,面临到需要配置环境的问题,单台服务器配置太麻烦,索性打算用docker来部署服务器环境。

本文只介绍如何在ubuntu16.04上正确安装nvidia-docker,对于概念的介绍会在结尾处贴出相关链接。安装nvidia-docker之前也需要安装docker,下面开始吧。

一、 docker安装:

1. 更换国内镜像源,推荐清华源或中科大源(本文选择的是中科大源)

1. sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak
2. sudo sed -i 's/archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
3. sudo apt update

2. 安装需要的依赖包

1. sudo apt install apt-transport-https ca-certificates software-properties-common curl

3. 添加GPG密钥,需指定源(仍然采用中科大源,采用默认的源下载速度慢,经常失败)

1. curl -fsSL https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
2. sudo add-apt-repository "deb [arch=amd64] https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu \
$(lsb_release -cs) stable"

4. 更新软件包

1. sudo apt update

5. 安装docker-ce

1. sudo apt install docker-ce

6. 测试docker安装成功

1. sudo docker run hello-world

若出现如下图所示则表示安装docker成功,离成功更近一步啦。

二、 nvidia-docker安装

1. 卸载nvidia-docker及其它GPU容器

1. docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
2. sudo apt-get purge -y nvidia-docker

2. 添加仓库包

1. curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
     sudo apt-key add -
2. distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
3. curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
4. sudo apt-get update

3. 安装nvidia-docker2

1. sudo apt-get install -y nvidia-docker2
2. sudo pkill -SIGHUP dockerd

4. 测试安装nvidia-docker成功(注意选择自己的cuda版本)

1. docker run --runtime=nvidia --rm nvidia/cuda:10.1-base nvidia-smi

若出现如下图所示则表示安装成功啦。


关于docker概念的介绍请参考:

老刘:看完此文,妈妈还会担心你docker入不了门?​zhuanlan.zhihu.com图标

docker官网:

Get Docker Engine - Community for Ubuntu​docs.docker.com图标

来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/31537575/viewspace-3137443/ ,如需转载,请注明出处,否则将追究法律责任。

下一篇: PostgreSQL的流复制
请登录后发表评论 登录
全部评论

注册时间:2018-05-16

  • 博文量
    4259
  • 访问量
    39841