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
commit 44b90df286
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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