allow multi gpu to be specified via ctr
Signed-off-by: Michael Crosby <michael@thepasture.io>
This commit is contained in:
		@@ -147,7 +147,7 @@ var (
 | 
			
		||||
			Name:  "pid-file",
 | 
			
		||||
			Usage: "file path to write the task's pid",
 | 
			
		||||
		},
 | 
			
		||||
		cli.IntFlag{
 | 
			
		||||
		cli.IntSliceFlag{
 | 
			
		||||
			Name:  "gpus",
 | 
			
		||||
			Usage: "add gpus to the container",
 | 
			
		||||
		},
 | 
			
		||||
 
 | 
			
		||||
@@ -250,7 +250,7 @@ func NewContainer(ctx gocontext.Context, client *containerd.Client, context *cli
 | 
			
		||||
			}))
 | 
			
		||||
		}
 | 
			
		||||
		if context.IsSet("gpus") {
 | 
			
		||||
			opts = append(opts, nvidia.WithGPUs(nvidia.WithDevices(context.Int("gpus")), nvidia.WithAllCapabilities))
 | 
			
		||||
			opts = append(opts, nvidia.WithGPUs(nvidia.WithDevices(context.IntSlice("gpus")...), nvidia.WithAllCapabilities))
 | 
			
		||||
		}
 | 
			
		||||
		if context.IsSet("allow-new-privs") {
 | 
			
		||||
			opts = append(opts, oci.WithNewPrivileges)
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user