diff --git a/tools/container/containerd/containerd.go b/tools/container/containerd/containerd.go index e5252e8c..2a41ec13 100644 --- a/tools/container/containerd/containerd.go +++ b/tools/container/containerd/containerd.go @@ -116,51 +116,40 @@ func main() { commonFlags := []cli.Flag{ &cli.StringFlag{ Name: "config", - Aliases: []string{"c"}, Usage: "Path to the containerd config file", Value: defaultConfig, Destination: &options.config, - EnvVars: []string{"CONTAINERD_CONFIG"}, + EnvVars: []string{"CONTAINERD_CONFIG", "RUNTIME_CONFIG"}, }, &cli.StringFlag{ Name: "socket", - Aliases: []string{"s"}, Usage: "Path to the containerd socket file", Value: defaultSocket, Destination: &options.socket, - EnvVars: []string{"CONTAINERD_SOCKET"}, - }, - &cli.StringFlag{ - Name: "runtime-class", - Aliases: []string{"r"}, - Usage: "The name of the runtime class to set for the nvidia-container-runtime", - Value: defaultRuntimeClass, - Destination: &options.runtimeClass, - EnvVars: []string{"CONTAINERD_RUNTIME_CLASS"}, - }, - &cli.StringFlag{ - Name: "runtime-type", - Usage: "The runtime_type to use for the configured runtime classes", - Value: defaultRuntmeType, - Destination: &options.runtimeType, - EnvVars: []string{"CONTAINERD_RUNTIME_TYPE"}, - }, - // The flags below are only used by the 'setup' command. - &cli.BoolFlag{ - Name: "set-as-default", - Aliases: []string{"d"}, - Usage: "Set nvidia-container-runtime as the default runtime", - Value: defaultSetAsDefault, - Destination: &options.setAsDefault, - EnvVars: []string{"CONTAINERD_SET_AS_DEFAULT"}, - Hidden: true, + EnvVars: []string{"CONTAINERD_SOCKET", "RUNTIME_SOCKET"}, }, &cli.StringFlag{ Name: "restart-mode", Usage: "Specify how containerd should be restarted; If 'none' is selected, it will not be restarted [signal | systemd | none]", Value: defaultRestartMode, Destination: &options.restartMode, - EnvVars: []string{"CONTAINERD_RESTART_MODE"}, + EnvVars: []string{"CONTAINERD_RESTART_MODE", "RUNTIME_RESTART_MODE"}, + }, + &cli.StringFlag{ + Name: "runtime-class", + Aliases: []string{"runtime-name", "nvidia-runtime-name"}, + Usage: "The name of the runtime class to set for the nvidia-container-runtime", + Value: defaultRuntimeClass, + Destination: &options.runtimeClass, + EnvVars: []string{"CONTAINERD_RUNTIME_CLASS", "NVIDIA_RUNTIME_NAME"}, + }, + &cli.BoolFlag{ + Name: "set-as-default", + Usage: "Set nvidia-container-runtime as the default runtime", + Value: defaultSetAsDefault, + Destination: &options.setAsDefault, + EnvVars: []string{"CONTAINERD_SET_AS_DEFAULT", "NVIDIA_RUNTIME_SET_AS_DEFAULT"}, + Hidden: true, }, &cli.StringFlag{ Name: "host-root", @@ -175,6 +164,13 @@ func main() { Destination: &options.useLegacyConfig, EnvVars: []string{"CONTAINERD_USE_LEGACY_CONFIG"}, }, + &cli.StringFlag{ + Name: "runtime-type", + Usage: "The runtime_type to use for the configured runtime classes", + Value: defaultRuntmeType, + Destination: &options.runtimeType, + EnvVars: []string{"CONTAINERD_RUNTIME_TYPE"}, + }, &cli.StringSliceFlag{ Name: "nvidia-container-runtime-modes.cdi.annotation-prefixes", Destination: &options.ContainerRuntimeModesCDIAnnotationPrefixes, diff --git a/tools/container/crio/crio.go b/tools/container/crio/crio.go index e0c7b87b..a55f005d 100644 --- a/tools/container/crio/crio.go +++ b/tools/container/crio/crio.go @@ -43,6 +43,7 @@ const ( // Config-based settings defaultConfig = "/etc/crio/crio.conf" + defaultSocket = "/var/run/crio/crio.sock" defaultRuntimeClass = "nvidia" defaultSetAsDefault = true defaultRestartMode = restartModeSystemd @@ -58,6 +59,7 @@ type options struct { runtimeDir string config string + socket string runtimeClass string setAsDefault bool restartMode string @@ -104,9 +106,53 @@ func main() { // only require the user to specify one set of flags for both 'startup' // and 'cleanup' to simplify things. commonFlags := []cli.Flag{ + &cli.StringFlag{ + Name: "config", + Usage: "Path to the cri-o config file", + Value: defaultConfig, + Destination: &options.config, + EnvVars: []string{"CRIO_CONFIG", "RUNTIME_CONFIG"}, + }, + &cli.StringFlag{ + Name: "socket", + Usage: "Path to the crio socket file", + Value: "", + Destination: &options.socket, + EnvVars: []string{"CRIO_SOCKET", "RUNTIME_SOCKET"}, + Hidden: true, + }, + &cli.StringFlag{ + Name: "restart-mode", + Usage: "Specify how cri-o should be restarted; If 'none' is selected, it will not be restarted [systemd | none]", + Value: defaultRestartMode, + Destination: &options.restartMode, + EnvVars: []string{"CRIO_RESTART_MODE", "RUNTIME_RESTART_MODE"}, + }, + &cli.StringFlag{ + Name: "runtime-class", + Aliases: []string{"runtime-name", "nvidia-runtime-name"}, + Usage: "The name of the runtime class to set for the nvidia-container-runtime", + Value: defaultRuntimeClass, + Destination: &options.runtimeClass, + EnvVars: []string{"CRIO_RUNTIME_CLASS", "NVIDIA_RUNTIME_NAME"}, + }, + &cli.BoolFlag{ + Name: "set-as-default", + Usage: "Set nvidia-container-runtime as the default runtime", + Value: defaultSetAsDefault, + Destination: &options.setAsDefault, + EnvVars: []string{"CRIO_SET_AS_DEFAULT", "NVIDIA_RUNTIME_SET_AS_DEFAULT"}, + Hidden: true, + }, + &cli.StringFlag{ + Name: "host-root", + Usage: "Specify the path to the host root to be used when restarting crio using systemd", + Value: defaultHostRootMount, + Destination: &options.hostRootMount, + EnvVars: []string{"HOST_ROOT_MOUNT"}, + }, &cli.StringFlag{ Name: "hooks-dir", - Aliases: []string{"d"}, Usage: "path to the cri-o hooks directory", Value: defaultHooksDir, Destination: &options.hooksDir, @@ -115,7 +161,6 @@ func main() { }, &cli.StringFlag{ Name: "hook-filename", - Aliases: []string{"f"}, Usage: "filename of the cri-o hook that will be created / removed in the hooks directory", Value: defaultHookFilename, Destination: &options.hookFilename, @@ -129,43 +174,7 @@ func main() { Destination: &options.configMode, EnvVars: []string{"CRIO_CONFIG_MODE"}, }, - &cli.StringFlag{ - Name: "config", - Usage: "Path to the cri-o config file", - Value: defaultConfig, - Destination: &options.config, - EnvVars: []string{"CRIO_CONFIG"}, - }, - &cli.StringFlag{ - Name: "runtime-class", - Usage: "The name of the runtime class to set for the nvidia-container-runtime", - Value: defaultRuntimeClass, - Destination: &options.runtimeClass, - EnvVars: []string{"CRIO_RUNTIME_CLASS"}, - }, // The flags below are only used by the 'setup' command. - &cli.BoolFlag{ - Name: "set-as-default", - Usage: "Set nvidia-container-runtime as the default runtime", - Value: defaultSetAsDefault, - Destination: &options.setAsDefault, - EnvVars: []string{"CRIO_SET_AS_DEFAULT"}, - Hidden: true, - }, - &cli.StringFlag{ - Name: "restart-mode", - Usage: "Specify how cri-o should be restarted; If 'none' is selected, it will not be restarted [systemd | none]", - Value: defaultRestartMode, - Destination: &options.restartMode, - EnvVars: []string{"CRIO_RESTART_MODE"}, - }, - &cli.StringFlag{ - Name: "host-root", - Usage: "Specify the path to the host root to be used when restarting crio using systemd", - Value: defaultHostRootMount, - Destination: &options.hostRootMount, - EnvVars: []string{"HOST_ROOT_MOUNT"}, - }, } // Update the subcommand flags with the common subcommand flags diff --git a/tools/container/docker/docker.go b/tools/container/docker/docker.go index 8f9747a4..942c3892 100644 --- a/tools/container/docker/docker.go +++ b/tools/container/docker/docker.go @@ -43,8 +43,9 @@ const ( defaultSocket = "/var/run/docker.sock" defaultSetAsDefault = true // defaultRuntimeName specifies the NVIDIA runtime to be use as the default runtime if setting the default runtime is enabled - defaultRuntimeName = nvidiaRuntimeName - defaultRestartMode = restartModeSignal + defaultRuntimeName = nvidiaRuntimeName + defaultRestartMode = restartModeSignal + defaultHostRootMount = "/host" reloadBackoff = 5 * time.Second maxReloadAttempts = 6 @@ -61,12 +62,13 @@ var nvidiaRuntimeBinaries = map[string]string{ // options stores the configuration from the command line or environment variables type options struct { - config string - socket string - runtimeName string - setAsDefault bool - runtimeDir string - restartMode string + config string + socket string + runtimeName string + setAsDefault bool + runtimeDir string + restartMode string + hostRootMount string } func main() { @@ -109,44 +111,50 @@ func main() { commonFlags := []cli.Flag{ &cli.StringFlag{ Name: "config", - Aliases: []string{"c"}, Usage: "Path to docker config file", Value: defaultConfig, Destination: &options.config, - EnvVars: []string{"DOCKER_CONFIG"}, + EnvVars: []string{"DOCKER_CONFIG", "RUNTIME_CONFIG"}, }, &cli.StringFlag{ Name: "socket", - Aliases: []string{"s"}, Usage: "Path to the docker socket file", Value: defaultSocket, Destination: &options.socket, - EnvVars: []string{"DOCKER_SOCKET"}, - }, - // The flags below are only used by the 'setup' command. - &cli.StringFlag{ - Name: "runtime-name", - Aliases: []string{"r"}, - Usage: "Specify the name of the `nvidia` runtime. If set-as-default is selected, the runtime is used as the default runtime.", - Value: defaultRuntimeName, - Destination: &options.runtimeName, - EnvVars: []string{"DOCKER_RUNTIME_NAME"}, - }, - &cli.BoolFlag{ - Name: "set-as-default", - Aliases: []string{"d"}, - Usage: "Set the `nvidia` runtime as the default runtime. If --runtime-name is specified as `nvidia-experimental` the experimental runtime is set as the default runtime instead", - Value: defaultSetAsDefault, - Destination: &options.setAsDefault, - EnvVars: []string{"DOCKER_SET_AS_DEFAULT"}, - Hidden: true, + EnvVars: []string{"DOCKER_SOCKET", "RUNTIME_SOCKET"}, }, &cli.StringFlag{ Name: "restart-mode", Usage: "Specify how docker should be restarted; If 'none' is selected it will not be restarted [signal | none]", Value: defaultRestartMode, Destination: &options.restartMode, - EnvVars: []string{"DOCKER_RESTART_MODE"}, + EnvVars: []string{"DOCKER_RESTART_MODE", "RUNTIME_RESTART_MODE"}, + }, + &cli.StringFlag{ + Name: "host-root", + Usage: "Specify the path to the host root to be used when restarting docker using systemd", + Value: defaultHostRootMount, + Destination: &options.hostRootMount, + EnvVars: []string{"HOST_ROOT_MOUNT"}, + // Restart using systemd is currently not supported. + // We hide this option for the time being. + Hidden: true, + }, + &cli.StringFlag{ + Name: "runtime-name", + Aliases: []string{"runtime-class", "nvidia-runtime-name"}, + Usage: "Specify the name of the `nvidia` runtime. If set-as-default is selected, the runtime is used as the default runtime.", + Value: defaultRuntimeName, + Destination: &options.runtimeName, + EnvVars: []string{"DOCKER_RUNTIME_NAME", "NVIDIA_RUNTIME_NAME"}, + }, + &cli.BoolFlag{ + Name: "set-as-default", + Usage: "Set the `nvidia` runtime as the default runtime. If --runtime-name is specified as `nvidia-experimental` the experimental runtime is set as the default runtime instead", + Value: defaultSetAsDefault, + Destination: &options.setAsDefault, + EnvVars: []string{"DOCKER_SET_AS_DEFAULT", "NVIDIA_RUNTIME_SET_AS_DEFAULT"}, + Hidden: true, }, }