UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-06-26 18:18:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add no-pivot option to toolkit

Browse Source 
Signed-off-by: Kathryn Baldauf <kabaldau@microsoft.com>
This commit is contained in:
Kathryn Baldauf 2020-01-31 16:43:41 -08:00
parent 2155c2d587
commit 5beddd6705
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nvidia-container-toolkit/hook_config.go
View File

@ -26,6 +26,7 @@ type CLIConfig struct {
Debug *string `toml:"debug"` Debug *string `toml:"debug"`
Ldcache *string `toml:"ldcache"` Ldcache *string `toml:"ldcache"`
LoadKmods bool `toml:"load-kmods"` LoadKmods bool `toml:"load-kmods"`
NoPivot bool `toml:"no-pivot"`
NoCgroups bool `toml:"no-cgroups"` NoCgroups bool `toml:"no-cgroups"`
User *string `toml:"user"` User *string `toml:"user"`
Ldconfig *string `toml:"ldconfig"` Ldconfig *string `toml:"ldconfig"`
@ -49,6 +50,7 @@ func getDefaultHookConfig() (config HookConfig) {
Debug: nil, Debug: nil,
Ldcache: nil, Ldcache: nil,
LoadKmods: true, LoadKmods: true,
NoPivot: false,
NoCgroups: false, NoCgroups: false,
User: nil, User: nil,
Ldconfig: nil, Ldconfig: nil,

3
nvidia-container-toolkit/main.go
View File

@ -101,6 +101,9 @@ func doPrestart() {
if cli.LoadKmods { if cli.LoadKmods {
args = append(args, "--load-kmods") args = append(args, "--load-kmods")
} }
if cli.NoPivot {
args = append(args, "--no-pivot")
}
if *debugflag { if *debugflag {
args = append(args, "--debug=/dev/stderr") args = append(args, "--debug=/dev/stderr")
} else if cli.Debug != nil { } else if cli.Debug != nil {