mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2025-06-16 11:30:20 +00:00
b4f0ad9c8e
The gated modifiers used to add support for GDS, Mofed, and CUDA Forward Comatibility only check the NVIDIA_VISIBLE_DEVICES envvar to determine whether GPUs are requested and modifications should be made. This means that use cases where volume mounts are used to request devices (e.g. when using the GPU Device Plugin) are not supported. This patch takes visibleDevicesFromEnvVar private, making VisibleDevices the only exported method to query valid devices. And edits the gated modifiers to use this func, ensuring device requests via mounts are also taken into acount. Signed-off-by: Carlos Eduardo Arango Gutierrez <eduardoa@nvidia.com> |
||
|---|---|---|
| .. | ||
| image | ||
| cli_test.go | ||
| cli.go | ||
| config_test.go | ||
| config.go | ||
| features.go | ||
| hook.go | ||
| runtime.go | ||
| toml_test.go | ||
| toml.go | ||
| toolkit-cli.go | ||