Adding a --load-kmods flag to the NVIDIA OCI hook

Signed-off-by: Jean Rouge <jer329@cornell.edu>
This commit is contained in:
Jean Rouge 2018-11-15 01:52:06 -08:00
parent a3c2f00c50
commit 90880078b9

View File

@ -90,6 +90,8 @@ func WithGPUs(opts ...Opts) oci.SpecOpts {
"oci-hook",
"--",
nvidiaPath,
// ensures the required kernel modules are properly loaded
"--load-kmods",
}, c.args()...),
Env: os.Environ(),
})