Merge pull request #2794 from wk8/wk8/nvidia_kmods

Adding a --load-kmods flag to the NVIDIA OCI hook
This commit is contained in:
Phil Estes
2018-11-16 11:50:37 +08:00
committed by GitHub

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(),
})