Ubuntu Docker 没有 GPU

引言

Docker 是一个开源的容器化平台，它允许开发者将应用程序及其依赖打包成一个独立的容器，然后部署到任何支持 Docker 的环境中。然而，很多开发者在使用 Docker 时可能会遇到一个问题，那就是 Docker 容器默认是没有 GPU 支持的。在本文中，我们将介绍为什么 Ubuntu Docker 默认没有 GPU 支持以及如何在 Ubuntu Docker 中启用 GPU 支持。

为什么 Ubuntu Docker 默认没有 GPU 支持

Ubuntu Docker 默认没有 GPU 支持是因为没有安装相应的驱动程序和库。GPU 是图形处理器的缩写，它主要用于加速图形和计算密集型任务。为了让 Docker 容器能够使用 GPU，我们需要在 Ubuntu Docker 中安装相应的驱动程序和库。

如何在 Ubuntu Docker 中启用 GPU 支持

要在 Ubuntu Docker 中启用 GPU 支持，我们需要执行以下步骤：

步骤 1：安装 NVIDIA 驱动程序

首先，我们需要在 Ubuntu Docker 中安装 NVIDIA 驱动程序。可以通过以下命令安装最新的 NVIDIA 驱动程序：

sudo apt-get update
sudo apt-get install nvidia-driver

安装完成后，可以使用以下命令来验证驱动程序是否正确安装：

nvidia-smi

如果成功显示 NVIDIA GPU 的信息，则表示驱动程序安装成功。

步骤 2：安装 Docker

接下来，我们需要在 Ubuntu Docker 中安装 Docker。可以使用以下命令安装 Docker：

sudo apt-get install docker.io

安装完成后，可以使用以下命令来验证 Docker 是否正确安装：

docker --version

如果成功显示 Docker 版本信息，则表示 Docker 安装成功。

步骤 3：安装 NVIDIA Docker 运行时

然后，我们需要安装 NVIDIA Docker 运行时，以便能够在 Docker 容器中使用 GPU。可以使用以下命令安装 NVIDIA Docker 运行时：

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L  | sudo apt-key add -
curl -s -L  | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

安装完成后，可以使用以下命令来验证 NVIDIA Docker 运行时是否正确安装：

docker run --gpus all nvidia/cuda:11.0-base nvidia-smi

如果成功显示 NVIDIA GPU 的信息，则表示 NVIDIA Docker 运行时安装成功。

步骤 4：启动带有 GPU 支持的 Docker 容器

最后，我们可以启动一个带有 GPU 支持的 Docker 容器，以验证 GPU 是否正常工作。可以使用以下命令启动一个带有 GPU 支持的 Docker 容器：

docker run --gpus all -it nvidia/cuda:11.0-base bash

然后，可以在 Docker 容器中执行以下命令来验证 GPU 是否正常工作：

nvidia-smi

如果成功显示 NVIDIA GPU 的信息，则表示 GPU 已经成功启用并能够在 Docker 容器中使用。

总结

在本文中，我们介绍了为什么 Ubuntu Docker 默认没有 GPU 支持以及如何在 Ubuntu Docker 中启用 GPU 支持。通过安装 NVIDIA 驱动程序、Docker 和 NVIDIA Docker 运行时，我们可以启用 GPU 支持，并在 Docker 容器中使用 GPU。这为开发者在 Docker 环境中进行图形和计算密集型任务提供了便利。

参考资料

  1. [NVIDIA Docker 官方文档](
  2. [NVIDIA Docker GitHub 仓库](