NVIDIA Tesla 系列 GPU(如 Tesla M40)支持两种工作模式:WDDM(Windows Display Driver Model) 和 TCC(Tesla Compute Cluster)。这两种模式的切换可以通过命令行工具 nvidia-smi
完成。
模式简介
WDDM 模式:
- 用于显示输出或桌面加速。
- 支持图形功能,如 DirectX。
- 默认用于工作站环境。
TCC 模式:
- 专为计算优化,无显示输出功能。
- 提高计算任务效率,降低延迟。
- 推荐用于高性能计算(HPC)、AI 训练/推理。
切换命令
在 Windows 系统中,使用以下命令切换模式:
1. 检查当前模式
运行以下命令以查看 Tesla GPU 的当前工作模式:
nvidia-smi -q -d COMPUTE
- 在输出中找到
Compute Mode
或TCC/WDDM
字段,显示当前模式。
2. 切换到 TCC 模式
执行以下命令切换到 TCC 模式:
nvidia-smi -g {GPU_ID} -dm 1
解释:
-g {GPU_ID}
:指定 GPU 的编号(从 0 开始)。用nvidia-smi
可以查看当前 GPU ID。-dm 1
:将模式设置为 TCC。
示例:
nvidia-smi -g 0 -dm 1
切换 GPU 0 到 TCC 模式。
3. 切换到 WDDM 模式
执行以下命令切换到 WDDM 模式:
nvidia-smi -g {GPU_ID} -dm 0
解释:
-dm 0
:将模式设置为 WDDM。
示例:
nvidia-smi -g 0 -dm 0
切换 GPU 0 到 WDDM 模式。
4. 重启系统
在完成模式切换后,通常需要重启系统才能生效。
注意事项
驱动版本:
- 确保安装了支持 TCC 模式的 NVIDIA 驱动程序。
- 专业卡(如 Tesla 系列)需要使用 NVIDIA 的数据中心驱动(NVIDIA Data Center Driver)。
支持限制:
- TCC 模式仅在专业卡(如 Tesla 系列)上支持,普通消费卡(如 GeForce 系列)无法切换到 TCC 模式。
- 某些情况下,Tesla 卡可能需要禁用显示输出才能启用 TCC 模式。
兼容性:
- 在运行 CUDA 应用程序时,建议使用 TCC 模式以提升性能。
权限:
- 运行命令需要管理员权限。