github/containerd
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

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

Browse Source 
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
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
contrib/nvidia/nvidia.go
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(),
}) })