Package org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu
package org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu
Package for GPU support classes.
-
ClassDescriptionIn addition to
GpuDevice, this include container id and more runtime information related to who is using the GPU device if possibleThis class is used to represent GPU device while allocation.This exception is to be thrown when allowed GPU device specification is empty or invalid.Factory to create GpuDocker Command Plugin instanceExecutes the "nvidia-smi" command and returns an object based on its output.Implementation to use nvidia-docker v1 as GPU docker command plugin.Implementation to use nvidia-docker v2 as GPU docker command plugin.