Adding a --load-kmods flag to the NVIDIA OCI hook
Signed-off-by: Jean Rouge <jer329@cornell.edu>
This commit is contained in:
parent
a3c2f00c50
commit
90880078b9
@ -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(),
|
||||
})
|
||||
|
Loading…
Reference in New Issue
Block a user