码迷,mamicode.com
首页 > 系统相关 > 详细

[Notes] Ubuntu安装nvidia-docker2

时间:2019-11-25 18:59:26      阅读:104      评论:0      收藏:0      [点我收藏+]

标签:finger   version   cat   common   agent   div   docker-ce   UNC   apt-get   

ubuntu版本为16.04.1 LTS (Xenial Xerus)。

 

本系统已安装nvidia驱动,但是未安装docker。

参考官方文档进行安装:https://docs.docker.com/install/linux/docker-ce/ubuntu/

然后在参考nvidia-docker github页进行安装:https://github.com/NVIDIA/nvidia-docker

一体化安装脚本:

sudo apt-get remove docker docker-engine docker.io containerd runc
sudo apt-get update
sudo apt-get install     apt-transport-https     ca-certificates     curl     gnupg-agent     software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
echo "chekc the fingureprint!"
sudo apt-key fingerprint 0EBFCD88
sudo add-apt-repository    "deb [arch=amd64] https://download.docker.com/linux/ubuntu    $(lsb_release -cs)    stable"
sudo apt-get update
sudo apt-get install docker-ce docker-ce-cli containerd.io
# install nvidia-docker2
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

  如果安装的docker版本比较新,那么nvidia-docke2可能就自带了,上面脚本的下半截可能就没啥用(从distribution开始)。

下面验证安装 成功,参考:https://github.com/NVIDIA/nvidia-docker

 

例如本机有两块显卡,运行下述验证:

docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

  

 

[Notes] Ubuntu安装nvidia-docker2

标签:finger   version   cat   common   agent   div   docker-ce   UNC   apt-get   

原文地址:https://www.cnblogs.com/immortalBlog/p/11929263.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!