UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-22 00:08:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add aliases for runtime-specific envvars

Browse Source 
This change ensures that the toolkit works with older
versions of the GPU Operator where runtime-specific envvars are
used to set options such as the config file location.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2024-10-18 11:19:07 +02:00
parent 38790c5df0
commit 1145ce2283
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/container/runtime/runtime.go
View File

@ -50,14 +50,14 @@ func Flags(opts *Options) []cli.Flag {
Usage: "Path to the runtime config file", Usage: "Path to the runtime config file",
Value: runtimeSpecificDefault, Value: runtimeSpecificDefault,
Destination: &opts.Config, Destination: &opts.Config,
EnvVars: []string{"RUNTIME_CONFIG"}, EnvVars: []string{"RUNTIME_CONFIG", "CONTAINERD_CONFIG", "DOCKER_CONFIG"},
}, },
&cli.StringFlag{ &cli.StringFlag{
Name: "socket", Name: "socket",
Usage: "Path to the runtime socket file", Usage: "Path to the runtime socket file",
Value: runtimeSpecificDefault, Value: runtimeSpecificDefault,
Destination: &opts.Socket, Destination: &opts.Socket,
EnvVars: []string{"RUNTIME_SOCKET"}, EnvVars: []string{"RUNTIME_SOCKET", "CONTAINERD_SOCKET", "DOCKER_SOCKET"},
}, },
&cli.StringFlag{ &cli.StringFlag{
Name: "restart-mode", Name: "restart-mode",
@ -79,14 +79,14 @@ func Flags(opts *Options) []cli.Flag {
Usage: "Specify the name of the `nvidia` runtime. If set-as-default is selected, the runtime is used as the default runtime.", Usage: "Specify the name of the `nvidia` runtime. If set-as-default is selected, the runtime is used as the default runtime.",
Value: defaultRuntimeName, Value: defaultRuntimeName,
Destination: &opts.RuntimeName, Destination: &opts.RuntimeName,
EnvVars: []string{"NVIDIA_RUNTIME_NAME"}, EnvVars: []string{"NVIDIA_RUNTIME_NAME", "CONTAINERD_RUNTIME_CLASS", "DOCKER_RUNTIME_NAME"},
}, },
&cli.BoolFlag{ &cli.BoolFlag{
Name: "set-as-default", Name: "set-as-default",
Usage: "Set the `nvidia` runtime as the default runtime.", Usage: "Set the `nvidia` runtime as the default runtime.",
Value: defaultSetAsDefault, Value: defaultSetAsDefault,
Destination: &opts.SetAsDefault, Destination: &opts.SetAsDefault,
EnvVars: []string{"NVIDIA_RUNTIME_SET_AS_DEFAULT"}, EnvVars: []string{"NVIDIA_RUNTIME_SET_AS_DEFAULT", "CONTAINERD_SET_AS_DEFAULT", "DOCKER_SET_AS_DEFAULT"},
Hidden: true, Hidden: true,
}, },
} }