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数量也可能会有所不同。
评论(0)