### 安装 NVIDIA 驱动插件并使 Docker 容器访问显卡资源 在 WSL2 的 Linux 环境中安装 NVIDIA 驱动插件，并确保 Docker 容器能够访问 NVIDIA 显卡资源，需要完成以下配置： #### 1. 安装 WSL2 和 NVIDIA 插件 首先，确保已安装 WSL2 并启用 NVIDIA 支持。可以通过以下命令安装带有 NVIDIA 支持的 Ubuntu-22.04 发行版： ```bash wsl --install -d Ubuntu-22.04 --nvidia ``` 此命令会自动安装 WSL2 和 NVIDIA 插件，确保 GPU 可用性[^1]。 验证 GPU 是否可用，可以在 Linux 系统中运行以下命令： ```bash nvidia-smi ``` 如果成功显示 GPU 信息，则说明 NVIDIA 驱动已正确安装。 #### 2. 配置 Docker 以支持 NVIDIA GPU 为了使 Docker 容器能够访问 NVIDIA 显卡资源，需在 Docker 中启用 NVIDIA 支持。可以按照以下方法进行配置： ##### 方法一：使用 `docker run` 命令直接启动容器 通过添加 `--gpus all` 参数，Docker 容器可以直接访问所有可用的 GPU 资源： ```bash docker run --gpus all nvidia/cuda:12.0-base nvidia-smi ``` 此命令将启动一个基于 NVIDIA CUDA 的容器，并验证 GPU 是否正常工作[^2]。 ##### 方法二：配置 Docker Daemon 若需要全局启用 GPU 支持，可修改 Docker 的守护进程配置文件 `/etc/docker/daemon.json`，添加以下内容： ```json { "runtimes": { "nvidia": { "path": "nvidia-container-runtime", "runtimeArgs": [] } }, "default-runtime": "nvidia" } ``` 保存后重启 Docker 服务以应用更改： ```bash sudo systemctl restart docker ``` 此后创建的所有容器都将默认支持 NVIDIA GPU[^2]。 #### 3. 在 Dockerfile 中配置环境变量 为了进一步优化容器内的 GPU 使用体验，可以在 Dockerfile 中设置相关环境变量。例如： ```dockerfile ENV NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 ``` 这些环境变量确保容器内的应用程序能够正确识别和利用 NVIDIA GPU[^2]。 #### 4. 验证容器内的 GPU 功能 进入已启用 GPU 支持的容器，运行以下命令以验证 GPU 是否正常工作： ```bash nvidia-smi ``` 如果成功显示 GPU 信息，则说明配置已完成[^1]。 --- ### 注意事项 - 确保系统已安装最新版本的 WSL2 和 Docker。 - 如果遇到问题，请检查 WSL2 和 NVIDIA 驱动是否为最新版本。 - 在多用户环境中，确保所有用户都具有正确的权限访问 GPU 资源。 ---