docker下查看显卡 信息

在 Docker 中查看显卡信息可以通过在容器中运行 nvidia-smi 命令来实现。

1. 首先需要在 Dockerfile 中添加 NVIDIA 显卡驱动，可以使用以下命令：

FROM nvidia/cuda:10.0-base

RUN apt-get update && apt-get install -y --no-install-recommends \
    nvidia-utils-410 \
    && rm -rf /var/lib/apt/lists/*

2. 构建 Docker 镜像并启动容器，在容器中运行 nvidia-smi 命令即可查看显卡信息：

docker build -t myimage .
docker run --gpus all myimage nvidia-smi

如果你的机器上有多张显卡，可以通过指定 GPU ID 来查看某一张显卡的信息：

docker run --gpus device=1 myimage nvidia-smi

其中 device=1 表示使用第二张显卡（ID 从 0 开始计数）。