Managing Data Operating System
Also available as:
PDF
loading table of contents...

Enable GPU Support for Docker on a non-Ambari Cluster

If you decide not to perform default installation and configuration of NVIDIA Docker plug-in, you must manually configure a non-Ambari cluster to enable GPU support for Docker.

  1. Go to the /etc/hadoop/conf/yarn-site.xml configuration file in your NodeManager.
  2. Configure the Nvidia Docker plugin for GPU:

    Property: yarn.nodemanager.resource-plugins.gpu.docker-plugin

    Default Value: nvidia-docker-v1

  3. Configure the Docker end point for nvidia-docker-plugin:

    Property: yarn.nodemanager.resource-plugins.gpu.docker-plugin.nvidia-docker-v1.endpoint

    Default Value: http://localhost:3476/v1.0/docker/cli

Sample configuration in your yarn-site.xml file:
<property>
   <name>yarn.nodemanager.resource-plugins.gpu.docker-plugin</name>
   <value>nvidia-docker-v1</value>
</property>  
  
<property>
   <name>yarn.nodemanager.resource-plugins.gpu.docker-plugin.nvidia-docker-v1.endpoint</name>
   <value>http://localhost:3476/v1.0/docker/cli</value>
</property>