204ca8230c
Introduce a device manager that manages the lifecycle of device plugins on the client. It fingerprints, collects stats, and forwards Reserve requests to the correct plugin. The manager, also handles device plugins failing and validates their output. |
||
|---|---|---|
| .. | ||
| cmd | ||
| nvml | ||
| device.go | ||
| device_test.go | ||
| fingerprint.go | ||
| fingerprint_test.go | ||
| README.md | ||
| stats.go | ||
| stats_test.go | ||
This package provides an implementation of nvidia device plugin
Behavior
Nvidia device plugin uses NVML bindings to get data regarding available nvidia devices and will expose them via Fingerprint RPC. GPUs can be excluded from fingerprinting by setting the ignored_gpu_ids field. Plugin sends statistics for fingerprinted devices every stats_period period.
Config
The configuration should be passed via an HCL file that begins with a top level config stanza:
config {
ignored_gpu_ids = ["uuid1", "uuid2"]
fingerprint_period = "5s"
}
The valid configuration options are:
ignored_gpu_ids(list(string):[]): list of GPU UUIDs strings that should not be exposed to nomadfingerprint_period(string:"5s"): The interval to repeat fingerprint process to identify possible changes.