ubuntu16.04 中安装 nvidia-docker2.0

最新推荐文章于 2024-09-10 11:36:23 发布

ddddfang

最新推荐文章于 2024-09-10 11:36:23 发布

阅读量1.4k

点赞数

CC 4.0 BY-SA版权

分类专栏：机器学习

本文链接：https://blog.youkuaiyun.com/ddddfang/article/details/90068856

机器学习专栏收录该内容

8 篇文章

订阅专栏

本文详细介绍了如何在Ubuntu环境下安装NVIDIA驱动、Docker-CE以及nvidia-docker2，实现GPU资源在Docker容器中的使用，包括安装步骤、测试方法及额外的配置技巧。

1.首先安装nvidia驱动

2.安装docker-ce(docker.io貌似不可以)

# 1
sudo apt-get update
sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common
# 2: 安装GPG证书
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
# 3: 写入软件源信息
sudo add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable"
# 4: 更新并安装 Docker-CE
sudo apt-get -y update
sudo apt-get -y install docker-ce

3. 安装 nvidia-docker2

# 执行命令 Add the package repositories
# command 1
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
 
# command 2
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
 
# command 3
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
 
# 正式安装 Install nvidia-docker2 and reload the Docker daemon configuration
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

4.测试

sudo docker run --runtime=nvidia --rm nvidia/cuda:8.0-devel nvidia-smi
sudo nvidia-docker run --rm nvidia/cuda:8.0-devel nvidia-smi
sudo docker run --runtime=nvidia --name cuda -ti -v/:/mnt nvidia/cuda:8.0-devel /bin/bash

或者直接下载ubuntu，然后在container里安装nvidia驱动，cuda+cudnn

ref to https://blog.youkuaiyun.com/quantum7/article/details/86416600

https://blog.youkuaiyun.com/y19930105/article/details/80763467

https://blog.youkuaiyun.com/qq_38079008/article/details/83620573