@InterfaceAudience.Private @InterfaceStability.Unstable public class GpuDiscoverer extends Object
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.YarnException
public void initialize(org.apache.hadoop.conf.Configuration config, NvidiaBinaryHelper nvidiaHelper) throws org.apache.hadoop.yarn.exceptions.YarnException
org.apache.hadoop.yarn.exceptions.YarnException
Copyright © 2008–2022 Apache Software Foundation. All rights reserved.