首页 > 云计算 > > Ubuntu16.04安装nvidia-docker
最近部门打算买一波服务器,面临到需要配置环境的问题,单台服务器配置太麻烦,索性打算用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 update2. 安装需要的依赖包
1. sudo apt install apt-transport-https ca-certificates software-properties-common curl3. 添加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 update5. 安装docker-ce
1. sudo apt install docker-ce6. 测试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-docker2. 添加仓库包
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 update3. 安装nvidia-docker2
1. sudo apt-get install -y nvidia-docker2
2. sudo pkill -SIGHUP dockerd4. 测试安装nvidia-docker成功(注意选择自己的cuda版本)
1. docker run --runtime=nvidia --rm nvidia/cuda:10.1-base nvidia-smi若出现如下图所示则表示安装成功啦。

关于docker概念的介绍请参考:
老刘:看完此文,妈妈还会担心你docker入不了门?
docker官网:
Get Docker Engine - Community for Ubuntu
来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/31537575/viewspace-3137443/ ,如需转载,请注明出处,否则将追究法律责任。