cuda.bindings.nvml.device_modify_drain_state#

cuda.bindings.nvml.device_modify_drain_state(intptr_t pci_info, int new_state)#

Modify the drain state of a GPU. This method forces a GPU to no longer accept new incoming requests. Any new NVML process will no longer see this GPU. Persistence mode for this GPU must be turned off before this call is made. Must be called as administrator. For Linux only.

Parameters:

pci_info (intptr_t) – The PCI address of the GPU drain state to be modified.
new_state (EnableState) – The drain state that should be entered, see nvmlEnableState_t.