GPUs in an AWS environment

Install GPU support on supported distributions on AWS

Using the Konvoy Image Builder, you can build an image that has support to use Nvidia GPU hardware to support GPU workloads. The DKP supported Nvidia driver versions is 470.x.

1. In your overrides/nvidia.yaml file add the following to enable GPU builds. You can also access and use the overrides repo.

   gpu:
     type: nvidia
   

2. Build your image using the following Konvoy image builder commands:

   konvoy-image build --region us-west-2 --source-ami=ami-12345abcdef images/ami/centos-7.yaml --overrides overrides/nvidia.yaml
   

   By default, your image builds in the us-west-2 region. To specify another region, set the --region flag:

   konvoy-image build --region us-east-1 --overrides override-source-ami.yaml images/ami/<Your OS>.yaml
   

   NOTE: Ensure that an AMI file is available for your OS selection.

   When the command is complete the ami id is printed and written to ./manifest.json.

   To use the built ami with Konvoy, specify it with the --ami flag when calling cluster create.

   dkp create cluster aws --cluster-name=$(whoami)-aws-cluster --region us-west-2 --ami <ami>
   

More helpful information can be found in the Nvidia Device Plug-in for Kubernetes instructions and the Installation Guide of Supported Platforms.

See also: Nvidia documentation