nvidia-smi
nvidia-smi -l 2
nvidia-smi --query-gpu=index,name,memory.total,memory.used,utilization.gpu,temperature.gpu,power.draw \
--format=csv,noheader,nounits
nvidia-smi --query-compute-apps=pid,process_name,used_memory --format=csv
nvidia-smi topo -m
nvidia-smi -q -d ECC,XID
CUDA 和多卡:
nvcc --version
python -c "import torch; print(torch.version.cuda, torch.cuda.is_available())"
nvidia-smi --query-gpu=name,compute_cap --format=csv
export CUDA_VISIBLE_DEVICES=0,1
容器和 Kubernetes:
docker run --rm --gpus all nvidia/cuda:12.4.0-base-ubuntu22.04 nvidia-smi
kubectl get nodes -o json | jq '.items[].status.allocatable["nvidia.com/gpu"]'
kubectl describe node <NODE> | grep -A10 'Allocated resources'
| 场景 | 命令 |
|---|---|
| 看显存占用 | nvidia-smi |
| 看进程 | nvidia-smi pmon -c 5 |
| 看拓扑/NVLink | nvidia-smi topo -m |
| 看 ECC/Xid | nvidia-smi -q -d ECC,XID |
| 压测 | gpu_burn 600 |
| 容器 GPU | --gpus all |
| K8s GPU | nvidia.com/gpu |
| 监控指标 | DCGM_FI_DEV_GPU_UTIL |
nvidia-smi 的 CUDA Version 是驱动支持上限,不是 runtime 版本。--gpus。nvidia-smi 有 PID 但进程已退出。