Add nvidia gpu support via libnvidia-container

This adds nvidia gpu support via the libnvidia-container project and
`nvidia-container-cli`.

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
This commit is contained in:
Michael Crosby
2018-03-28 16:40:27 -04:00
parent 544557289a
commit b949697a9c
4 changed files with 199 additions and 1 deletions

View File

@@ -93,6 +93,10 @@ var ContainerFlags = []cli.Flag{
Name: "pid-file",
Usage: "file path to write the task's pid",
},
cli.IntFlag{
Name: "gpus",
Usage: "add gpus to the container",
},
}
func loadSpec(path string, s *specs.Spec) error {