前一阵子写了一篇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
![2c9693e9f405d1668f1ffaa68c561c0f.png](https://i-blog.csdnimg.cn/blog_migrate/8207371f1489d331a74f61de60c95e78.jpeg)
根据nvidia-docker的官方github,还需要安装nvidia-container-toolkit
# Add the p