Package org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu

Class NvidiaDockerV2CommandPlugin

java.lang.Object

org.apache.hadoop.yarn.server.nodemanager.containermanager.resourceplugin.gpu.NvidiaDockerV2CommandPlugin

All Implemented Interfaces:

DockerCommandPlugin

public class NvidiaDockerV2CommandPlugin
extends Object
implements DockerCommandPlugin

Implementation to use nvidia-docker v2 as GPU docker command plugin.

Constructor Summary

Constructors

Constructor	Description
NvidiaDockerV2CommandPlugin()

Method Summary

Modifier and Type	Method	Description
DockerVolumeCommand	getCleanupDockerVolumesCommand(Container container)	Cleanup volumes created for one docker container
DockerVolumeCommand	getCreateDockerVolumeCommand(Container container)	Create volume when needed.
protected boolean	requestsGpu(Container container)
void	updateDockerRunCommand(DockerRunCommand dockerRunCommand, Container container)	Update docker run command

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

NvidiaDockerV2CommandPlugin

public NvidiaDockerV2CommandPlugin()

Method Details

requestsGpu

@VisibleForTesting
protected boolean requestsGpu(Container container)

updateDockerRunCommand

public void updateDockerRunCommand(DockerRunCommand dockerRunCommand,
 Container container)
                            throws ContainerExecutionException

Description copied from interface: DockerCommandPlugin

Update docker run command

Specified by:

updateDockerRunCommand in interface DockerCommandPlugin

Parameters:

dockerRunCommand - docker run command

container - NM container

Throws:

ContainerExecutionException - if any issue occurs

getCreateDockerVolumeCommand

public DockerVolumeCommand getCreateDockerVolumeCommand(Container container)
                                                 throws ContainerExecutionException

Description copied from interface: DockerCommandPlugin

Create volume when needed.

Specified by:

getCreateDockerVolumeCommand in interface DockerCommandPlugin

Parameters:

container - container

Returns:

DockerVolumeCommand to create volume

Throws:

ContainerExecutionException - when any issue happens

getCleanupDockerVolumesCommand

public DockerVolumeCommand getCleanupDockerVolumesCommand(Container container)
                                                   throws ContainerExecutionException

Description copied from interface: DockerCommandPlugin

Cleanup volumes created for one docker container

Specified by:

getCleanupDockerVolumesCommand in interface DockerCommandPlugin

Parameters:

container - container

Returns:

DockerVolumeCommand to remove volume

Throws:

ContainerExecutionException - when any issue happens