如何查看NVIDIA Container Toolkit是否配置成功

要确认 NVIDIA Container Toolkit 是否已成功配置，可以按照以下步骤进行检查：

首先，确保你的系统已经正确安装了 NVIDIA 驱动程序，并且可以识别你的 GPU。你可以使用 nvidia-smi 命令来进行检查：

复制代码

nvidia-smi

这应该会显示你的 GPU 信息和当前的 GPU 使用情况。如果这个命令失败，说明你的 NVIDIA 驱动程序可能没有正确安装。

确保你已经安装了 Docker，并且 Docker 服务正在运行：

复制代码

docker --version

检查 Docker 服务状态：

复制代码

sudo systemctl status docker

确保 NVIDIA Container Toolkit 已正确安装。你可以使用以下命令来检查 nvidia-container-runtime 的版本：

复制代码

nvidia-container-runtime --version

确保 Docker 已配置为使用 NVIDIA Container Toolkit。检查 Docker 的 daemon.json 文件，通常位于 /etc/docker/daemon.json，并确保它包含以下配置：

复制代码

{
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}

如果 daemon.json 文件不存在，你可以创建它并添加上述配置。然后重新启动 Docker 服务：

复制代码

sudo systemctl restart docker

运行一个基于 nvidia/cuda 镜像的测试容器，以确认 NVIDIA Container Toolkit 已正确配置。以下命令将会启动一个容器，并在其中运行 nvidia-smi：

复制代码

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

你应该看到类似于在主机上运行 nvidia-smi 的输出，显示 GPU 的详细信息。如果成功，说明 NVIDIA Container Toolkit 配置正确。

你也可以检查 Docker 是否已正确识别到 nvidia 运行时。运行以下命令查看 Docker 支持的运行时列表：

复制代码

docker info | grep Runtimes

你应该看到 nvidia 运行时在列表中。

如果遇到问题，可以查看 NVIDIA Container Toolkit 的日志进行诊断。日志文件通常位于 /var/log/nvidia-container-runtime.log 或者 Docker 的日志文件中。

通过以上步骤，你应该能够确认 NVIDIA Container Toolkit 是否已成功配置。如果在某一步遇到问题，仔细检查相关配置和日志，通常可以找到解决方法。