将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标志。