docker nvidia

Docker 19.03版本提供了对NVIDIA显卡的支持，使得在Docker容器中使用GPU变得更加方便，本文将详细介绍如何在Docker 19.03中使用NVIDIA显卡。

你需要确保你的系统已经安装了NVIDIA显卡驱动，并且Docker服务也已经启动，如果你的系统中没有NVIDIA显卡，那么你可能无法使用Docker的GPU支持。

接下来，你需要安装NVIDIA Container Toolkit，这是一个用于在Docker容器中使用GPU的工具包，你可以在NVIDIA的官方网站上下载，安装完成后，你需要重启Docker服务。

在安装完NVIDIA Container Toolkit后，你就可以在Docker中使用GPU了，你可以通过以下命令来检查是否已经成功启用了GPU支持：

docker run --rm nvidia/cuda:10.2-base nvidia-smi 

这个命令会运行一个包含CUDA 10.2的基础镜像的Docker容器，并显示关于GPU的信息，如果你看到了关于GPU的信息，那么就说明你已经成功启用了GPU支持。

在使用Docker时，你可以通过在运行容器时添加`–gpus all`参数来启用所有可用的GPU。

docker run --gpus all -it ubuntu nvidia-smi 

这个命令会运行一个Ubuntu容器，并使用所有可用的GPU，在这个容器中，你可以运行任何需要GPU的程序。

除了使用`–gpus all`参数外，你还可以使用`–gpus `参数来指定使用特定数量的GPU，如果你想使用两个GPU，你可以运行：

docker run --gpus '"2"' -it ubuntu nvidia-smi 

这个命令会运行一个Ubuntu容器，并使用两个GPU，在这个容器中，你可以运行需要GPU的程序。

Docker 19.03提供了对NVIDIA显卡的支持，使得在Docker容器中使用GPU变得更加方便，你只需要安装NVIDIA Container Toolkit，并在运行容器时添加相应的参数，就可以在Docker中使用GPU了。

相关问题与解答**

问题1：我安装了NVIDIA Container Toolkit，但是仍然无法在Docker中使用GPU，怎么办？

答：请检查你是否已经正确安装了NVIDIA Container Toolkit，以及是否正确重启了Docker服务，如果问题仍然存在，你可以尝试卸载并重新安装NVIDIA Container Toolkit。

问题2：我在运行Docker容器时使用了`–gpus all`参数，但是我看到的GPU信息和预期的不同，为什么？

答：`–gpus all`参数会启用所有可用的GPU，如果你只看到一部分GPU的信息，可能是因为其他GPU没有被正确识别或者已经被其他程序占用，你可以尝试运行`nvidia-smi`命令来查看所有的GPU信息。

问题3：我在运行Docker容器时使用了`–gpus `参数，但是我指定的GPU数量和我实际使用的GPU数量不一致，为什么？

答：`–gpus `参数只是指定了你希望Docker使用多少个GPU，实际上，Docker可能会根据你的硬件配置和使用情况动态调整使用的GPU数量，即使你指定了特定的GPU数量，实际使用的GPU数量也可能会有所不同。