将 cuda 计算模式切换到默认模式

Question

我使用nvidia-smi查看计算节点上每个GPU的状态，但发现其中一个是E. Thread。有什么简单的方法可以将其切换回default mode? ------------------------------------------------------+
| NVIDIA-SMI 346.46 Driver Version: 346.46 |
|-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 GeForce GTX 680 Off | 0000:03:00.0 N/A | N/A | | 30% 30C P8 N/A / N/A | 10MiB / 4095MiB | N/A E. Thread | +-------------------------------+----------------------+----------------------+ | 1 GeForce GTX 680 Off | 0000:04:00.0 N/A | N/A | | 30% 29C P8 N/A / N/A | 10MiB / 4095MiB | N/A Default | +-------------------------------+----------------------+----------------------+

Answer 1

nvidia-smi -h

会给你命令行帮助，还有一个手册页：

man nvidia-smi

以下命令应该将该设备重置为计算模式 0（默认）

nvidia-smi -i 0 -c 0

应该将该设备 (0) 重置为计算模式 0（默认）

您需要具有root权限才能以这种方式修改设备，因此要么是root用户，要么使用sudo运行命令

Answer 2

为了知道当前的Compute Mode，使用下面的命令

nvidia-smi  --query | grep 'Compute Mode'

为了设置Compute Mode，使用以下命令

sudo nvidia-smi -c $i

其中 $i 可以是 0、1、2 或 3。这些数字的含义如下 -

0 Default
1 Exclusive_Thread
2 Prohibited 
3 Exclusive_Process

回答你的问题

sudo nvidia-smi -c 0

这会将所有 GPU 设置为默认模式。如果您只想将其应用于 GPU 0，请使用 -i 0 标志。