mirror of
https://github.com/hanzoai/visor.git
synced 2026-09-24 17:04:14 +00:00
The --nvproxy flag allows container GPU usage to be specified via device nodes and mounts provided in the runtime spec, as when using Kubernetes with GKE's Nvidia GPU device plugin (https://github.com/GoogleCloudPlatform/container-engine-accelerators/tree/master/cmd/nvidia_gpu). The --nvproxy-docker flag additionally allows container GPU usage to be specified via the NVIDIA_VISIBLE_DEVICES container environment variable, as when using `docker --gpus`. This does not require the Nvidia Container Toolkit (or the Nvidia Container Runtime [Hook], which are part of the Toolkit), but does require libnvidia-container, which is typically installed as a dependency of the Nvidia Container Toolkit. Updates #14 PiperOrigin-RevId: 535002602