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

<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>