docker用gpu的参数_初探Docker调用GPU

前一阵子写了一篇docker的学习笔记^[1]，但是当时没有gpu，所以没法做显卡调用相关的内容。最近机房的电脑启动了，有了实验环境，打算把docker调用gpu相关的内容测试一下。实验环境依然为Ubuntu16.04。

根据所看到的资料^[2]以及docker的release notes^[3]显示，19年7月的时候，docker已经在19.03版本中加入了NVIDIA GPU的runtime，所以本篇会针对此进行学习和实验，而不是旧版的nvidia-docker或者nvidia-docker2。

环境安装

首先需要在拥有NVIDIA GPU显卡的机子上安装显卡驱动，使用一些参数会让安装省去许多麻烦

./NVIDIA-Linux-x86_64-440.36.run --no-opengl-files --no-x-check --no-nouveau-check

然后根据docker的官方文档在服务器中安装docker-ce，安装完后默认可以得到最新版的docker，目前是19.03.6

根据nvidia-docker的官方github，还需要安装nvidia-container-toolkit

# Add the package repositories
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.listsudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

测试

我在单卡服务器上测试了官方示例，结果如图

为了能在docker中训练与部署模型，我们需要一个开发环境的镜像。pytorch在DockerHub^[4]上传了许多已经制作好的镜像，我们可以用来作为基础镜像

docker pull pytorch/pytorch:1.1.0-cuda10.0-cudnn7.5-devel

映射个测试目录，启动容器

docker run -it --gpus=all -v /home/eagle/project/test:/workspace pytorch/pytorch:1.1.0-cuda10.0-cudnn7.5-devel

在测试目录/home/eagle/project/test下创建一个名为test.py的文件，随便找一份训练代码写进去，在容器中运行python test进行测试

可以看到训练正常，至此测试就可以结束了。

其它

可以看到，docker 19.03之后，调用NVIDIA GPU还是很方便的。不过目前NVIDIA官方暂未支持docker-compose^[5]，期待后续的完善。

PS：期待WSL2也尽快的支持GPU调用，这样炼丹就会更方便了。

参考

^Docker学习笔记 - 知乎 https://zhuanlan.zhihu.com/p/105818436
^K8S 生态周报| 2019-07-21~2019-07-28 - 知乎 https://zhuanlan.zhihu.com/p/75591755
^Docker Engine release notes | Docker Documentation https://docs.docker.com/engine/release-notes/#19030
^pytorch/pytorch Tags - Docker Hub https://hub.docker.com/r/pytorch/pytorch/tags
^Support for NVIDIA GPUs under Docker Compose · Issue #6691 · docker/compose https://github.com/docker/compose/issues/6691

docker用gpu的参数_初探Docker调用GPU相关推荐

docker用gpu的参数_从零开始入门 K8s | GPU 管理和 Device Plugin 工作机制
导读:2016 年,随着 AlphaGo 的走红和 TensorFlow 项目的异军突起,一场名为 AI 的技术革命迅速从学术圈蔓延到了工业界,所谓 AI 革命从此拉开了帷幕.该热潮的背后推手正是云计 ...
docker配置容器mysql启动参数_修改Docker容器启动配置参数
有时候,我们创建容器时忘了添加参数 --restart=always ,当 Docker 重启时,容器未能自动启动, 现在要添加该参数怎么办呢,方法有二: 1.Docker 命令修改 docker c ...
docker 查看容器磁盘大小_软件安全 : Docker逃逸详解
前言在虚拟化技术日益成熟的今天,Docker作为必须使用的一项服务,它的安全性尤为重要.今天,我们来一起学习Docker最大的安全威胁--逃逸技术. Docker是什么 Docker 是一个开源的应 ...
jar docker镜像不能启线程_使用docker部署SpringCloud项目
点击上方"Java知音",选择"置顶公众号" 技术文章第一时间送达! 作者:现实太狗ochttps://blog.csdn.net/u011508407/ 分类 ...
docker部署python web应用_安装docker并部署web项目
一.docker简介 1.docker定义:docker是一个用来装应用的容器,就像杯子可以装水,笔筒可以装笔,书包可以放书一样.你可以把"hello world!"放到docke ...
docker容易内部怎么编辑_在Docker的工作流中常见问题及最终方案
我们用Python代码演示,代码都是共通的大家都要了解下. 这次我们创建一个Hello world的web服务器. 一 mkdir -p identidock/app #首先创建一个新的multiid ...
docker image设置jdk版本_使用Docker搭建Java环境的步骤方法
Docker是干什么的 Docker 是一个基于Linux容器(LXC-linux container)的高级容器引擎,基于go语言开发,源代码托管在 Github 上, 遵从Apache2.0协议开 ...
docker部署python web应用_使用docker一键部署Web应用
前几天借着部署一个小系统的机会尝试了一把docker,把Web应用,数据库,nginx统统都用docker管理.把打包好的代码扔到服务器上一键就部署好了,不需要考虑服务器的环境问题,简直不要太爽.但感 ...
docker搭建java测试环境_基于docker构建测试环境
目录 0x01介绍 0x02 镜像基本操作 0x03 容器基本操作 0x04 容器的修改与保存 0x05 使用Dockerfile定制镜像 0x01介绍 Docker 是一个开源的应用容器引擎,基于 ...

docker用gpu的参数_初探Docker调用GPU

环境安装

测试

其它

参考

docker用gpu的参数_初探Docker调用GPU相关推荐

最新文章

热门文章