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