Skip to content

Latest commit

 

History

History
49 lines (38 loc) · 1.33 KB

nvidia-gpu.md

File metadata and controls

49 lines (38 loc) · 1.33 KB
Raw

Nvidia GPU

konvoy-image-builder supports building images with installed Nvidia host components that can be used to run GPU workloads on machines that come with a dedicated GPU hardware.

To enable GPU build, add the following override:

---
gpu:
  types:
    - nvidia
build_name_extra: "-nvidia"

There is an existing prepared override file in overrides/nvidia.yaml that will enable installing Nvidia host components to the built image.

Example:

build --region us-west-2 --source-ami=ami-12345abcdef images/ami/centos-79.yaml \
    --overrides overrides/nvidia.yaml \
    --instance-type=g4dn.2xlarge

Example of building on AWS with Ubuntu 20.04 as the operating system and p2.xlarge as the AWS machines you are going to use:

konvoy-image build aws --region us-west-2 images/ami/ubuntu-2004.yaml \
    --overrides overrides/nvidia.yaml \
    --instance-type=p2.xlarge

NOTE: Providing the instance type is required for building GPU images.

Supported images

base os nvidia
centos-7 ✅
rhel-79 ✅
rhel-84 ✅
rhel-86 ✅
ubuntu-18 ✅
ubuntu-20 ✅
flatcar ✅
sles-15 ✅