DC/OS supports allocating GPUs (Graphics Processing Units) to your long-running DC/OS services. Adding GPUs to your services can dramatically accelerate big data workloads.
With GPU-based scheduling, you can share cluster resources for traditional and machine learning workloads, as well as dynamically allocate GPU resources inside those clusters and free them when needed. You can reserve GPU resources for the workloads that need them, or pool these GPU-enabled resources with the rest of the infrastructure for higher overall utilization.
After installing DC/OS with GPUs enabled, you can specify GPUs in your application definitions with the
Installing DC/OS with GPUs Enabled
GPUs must be enabled during DC/OS installation. Follow the instructions below to enable GPUs based on your specific DC/OS deployment method.
Custom DC/OS Installation with GPUs
Install the NVIDIA Management Library (NVML) on each node of your cluster that has GPUs. The minimum required NVIDIA driver version is 340.29. For detailed installation instructions, see the Mesos GPU support documentation.
Install DC/OS using the custom advanced installation instructions. Here are the GPU-specific configuration parameters:
enable_gpu_isolation: Indicates whether to enable GPU support in DC/OS. By default, this is set to
gpus_are_scarce: Indicates whether to treat GPUs as a scarce resource in the cluster. By default, this is set to
gpus_are_scarce: 'true', which means DC/OS reserves GPU nodes exclusively for services that are configured to consume GPU resources. It’s important to note that this setting will influence which agent nodes a GPU-aware framework will be deployed on DC/OS. This setting does not influence the individual tasks which the frameworks might launch while the framework is running. It is possible for a framework to schedule a non-GPU task on an agent node where GPU’s are present.
AWS EC2 DC/OS Installation with GPUs
- The AWS DC/OS advanced template system requirements.
zen.shscript copied to your local machine. The script and instructions are here.
- Run the
zen.shscript to create the Zen template dependencies. These dependencies will be used as input to create your stack in CloudFormation.
bash ./zen.sh <stack-name>
Follow the instructions here to create a cluster with advanced AWS templates, using the following GPU-specific configuration.
On the Create Stack > Specify Details page, specify your stack information and click Next. Here are the GPU-specific settings.
CustomAMI - Specify the custom AMI for your region:
MasterInstanceType - Accept the default master instance type (For example,
PrivateAgentInstanceType - Specify an AWS GPU machine type (For example,
PublicAgentInstanceType - Specify an AWS GPU machine type (For example,
On the Options page, accept the defaults and click Next. You can choose whether to roll back on failure. By default this option is set to Yes.
On the Review page, check the acknowledgement box, then click Create.