| 用途 | 为容器挂载 GPU 设备 |
|---|---|
| 语法 | --gpus all / --gpus 1 / --gpus '"device=0,2"'(设备列表写法) |
| 前置 | NVIDIA 驱动 + nvidia-container-toolkit |
示例
示例
docker run --rm --gpus all nvidia/cuda:12.4.0-base-ubuntu22.04 nvidia-smi
docker run --rm --gpus 1 python:3.12-slim nvidia-smi || true该参数依赖 NVIDIA 容器运行时;Windows 环境经 WSL2 后端支持,纯 CPU 机器上会报错。