设置Docker以使用与Nvidia GPU兼容的容器[Ubuntu 18.04]

2019年11月18日2020年1月15日

从19.03版开始，Docker就对“使用Nvidia GPU进行容器加速”提供了本地支持。使用NVIDIA Container Toolkit可以使用此功能。在此进行所需的初始设置。

环境
操作系统：Ubuntu 18.04 LTS
显卡：GTX1080

步骤XNUMX：检查先决条件

在使用Docker的“通过GPU加速容器”之前，让我们检查一下环境。GPU容器不能在过旧的环境中使用。

• GNU / Linux x86_64（内核版本> 3.10）←Ubuntu 18.04很好
• 具有架构的NVIDIA GPU> Fermi（2.1）←如果GPU用于台式机，则可以使用GTX 600或更高版本
• NVIDIA驱动程序版本≧361.93←参见下面的步骤XNUMX

有关详细信息，请参见以下页面。

步骤XNUMX：安装Docker

如果尚未安装Docker，请按照以下步骤进行安装。

步骤XNUMX：安装Nvidia GPU驱动程序

Nvidia驱动程序是运行GPU容器（CUDA容器）所必需的。

Nvidia驱动程序的版本应大于您要使用的CUDA的版本（例如，在tensorflow中，所需的CUDA版本会根据所使用的版本而完全不同）。

只要正确安装了Nvidia驱动程序，就不需要单独安装CUDA工具包或NVIDIA Container Toolkit。

Nvidia驱动程序的安装方法概述如下。

步骤XNUMX：在Docker中使GPU容器可用

从终端输入以下内容。
官方参考： https://github.com/NVIDIA/nvidia-docker

＃添加软件包存储库分配= $（./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更新&& sudo apt-get install -y nvidia-container-toolkit sudo systemctl重新启动docker

检查操作

＃使用最新的官方CUDA镜像docker run --gpus测试所有nvidia / cuda：9.0-base nvidia-smi

如果一切顺利，您将看到此显示。

要从现在开始使用GPU容器，请将“ –gpus”选项添加到“ docker run”。

如果您只是安装了docker，则除非您在docker运行之前添加sudo，否则它可能无法运行。
[Ubuntu]在不使用sudo命令的情况下运行docker的设置和注意事项

这些是在Ubuntu上使用Docker的GPU兼容容器所需的初始设置。

参考书