From 9c01fe20d31d3120c8bd2e0e68c411a3ebeabc64 Mon Sep 17 00:00:00 2001 From: ktock Date: Mon, 14 Jun 2021 23:58:22 +0900 Subject: [PATCH] Allow to pass --no-cgroup option to nvidia-container-cli Signed-off-by: Kohei Tokunaga --- contrib/nvidia/nvidia.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/contrib/nvidia/nvidia.go b/contrib/nvidia/nvidia.go index 6a351771f..d08044aa2 100644 --- a/contrib/nvidia/nvidia.go +++ b/contrib/nvidia/nvidia.go @@ -111,6 +111,7 @@ type config struct { LDConfig string Requirements []string OCIHookPath string + NoCgroups bool } func (c *config) args() []string { @@ -137,6 +138,9 @@ func (c *config) args() []string { for _, r := range c.Requirements { args = append(args, fmt.Sprintf("--require=%s", r)) } + if c.NoCgroups { + args = append(args, "--no-cgroups") + } args = append(args, "--pid={{pid}}", "{{rootfs}}") return args } @@ -209,3 +213,9 @@ func WithLookupOCIHookPath(name string) Opts { return nil } } + +// WithNoCgroups passes --no-cgroups option to nvidia-container-cli. +func WithNoCgroups(c *config) error { + c.NoCgroups = true + return nil +}