blakeblackshear / blakeshome-charts

Repo for helm charts
41 stars 52 forks source link

Add NVIDIA GPU Support #22

Closed maxirus closed 2 years ago

maxirus commented 2 years ago

In order to enable support NVIDIA GPU, a number of configuration changes must be implemented which (most) are accessible now via resources and env properties. However, not all nodes make nvidia the default runtimeClass and this is currently not configurable.

Instead, a boolean flag should exist to automate the injection of NVIDIA_DRIVER_CAPABILITIES & NVIDIA_VISIBLE_DEVICES env vars as well as the "nvidia.com/gpu": 1 resource limit. In addition, an option runtimeClass property should be exposed.