diff --git a/nvidia-container-toolkit/hook_config.go b/nvidia-container-toolkit/hook_config.go index c8e10388..320b842d 100644 --- a/nvidia-container-toolkit/hook_config.go +++ b/nvidia-container-toolkit/hook_config.go @@ -26,6 +26,7 @@ type CLIConfig struct { Debug *string `toml:"debug"` Ldcache *string `toml:"ldcache"` LoadKmods bool `toml:"load-kmods"` + NoPivot bool `toml:"no-pivot"` NoCgroups bool `toml:"no-cgroups"` User *string `toml:"user"` Ldconfig *string `toml:"ldconfig"` @@ -49,6 +50,7 @@ func getDefaultHookConfig() (config HookConfig) { Debug: nil, Ldcache: nil, LoadKmods: true, + NoPivot: false, NoCgroups: false, User: nil, Ldconfig: nil, diff --git a/nvidia-container-toolkit/main.go b/nvidia-container-toolkit/main.go index 52f4c0ef..010ff359 100644 --- a/nvidia-container-toolkit/main.go +++ b/nvidia-container-toolkit/main.go @@ -101,6 +101,9 @@ func doPrestart() { if cli.LoadKmods { args = append(args, "--load-kmods") } + if cli.NoPivot { + args = append(args, "--no-pivot") + } if *debugflag { args = append(args, "--debug=/dev/stderr") } else if cli.Debug != nil {