将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(默认)

您需要具有根用户权限才能以这种方式修改设备，因此要么是根用户，要么使用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设置为默认模式。如果只想将其应用于GPU0，请使用-i 0标志。