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

switch cuda compute mode to default mode

我使用nvidia-smi查看计算节点上每个GPU的状态,但发现其中一个是E. Thread。有什么简单的方法可以将其切换回 default mode? ---------------------------------------------- ----+<br> | NVIDIA-SMI 346.46 驱动程序版本:346.46 |<br> |----------------------------+---------------- -----+------------------------+ | GPU 名称持久性-M|总线 ID Disp.A |挥发性Uncorr。纠错码 | |风扇温度性能功率:Usage/Cap|内存使用 | GPU-Util 计算 M。 |===============================+================= =====+======================| | 0 GeForce GTX 680 关闭 | 0000:03:00.0 N/A | N/A | | 30% 30C P8 N/A / N/A | 10MiB / 4095MiB | N/A E.线程 | +----------------------------+---------------- -----+------------------------+ | 1 GeForce GTX 680 关闭 | 0000:04:00.0 N/A | N/A | | 30% 29C P8 N/A / N/A | 10MiB / 4095MiB | N/A 默认 | +----------------------------+---------------- -----+------------------------+

nvidia-smi -h

会给你命令行帮助,还有一个手册页:

man nvidia-smi

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

nvidia-smi -i 0 -c 0

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

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

命令

为了知道当前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 标志。