@InterfaceAudience.Private
 @InterfaceStability.Unstable
public class GpuDiscoverer
extends org.apache.hadoop.conf.Configured
| Modifier and Type | Field and Description | 
|---|---|
| static org.slf4j.Logger | LOG | 
| Constructor and Description | 
|---|
| GpuDiscoverer() | 
| Modifier and Type | Method and Description | 
|---|---|
| GpuDeviceInformation | getGpuDeviceInformation()Get GPU device information from system. | 
| List<GpuDevice> | getGpusUsableByYarn()Get list of GPU devices usable by YARN. | 
| void | initialize(org.apache.hadoop.conf.Configuration config,
          NvidiaBinaryHelper nvidiaHelper) | 
public GpuDeviceInformation getGpuDeviceInformation() throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnException - when any error happenspublic List<GpuDevice> getGpusUsableByYarn() throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnExceptionpublic void initialize(org.apache.hadoop.conf.Configuration config,
                       NvidiaBinaryHelper nvidiaHelper)
                throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnExceptionCopyright © 2008–2023 Apache Software Foundation. All rights reserved.