@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.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–2023 Apache Software Foundation. All rights reserved.