From 90880078b90c803929f35a768ff47ec2d84b3f1f Mon Sep 17 00:00:00 2001
From: Jean Rouge <jer329@cornell.edu>
Date: Thu, 15 Nov 2018 01:52:06 -0800
Subject: [PATCH] Adding a --load-kmods flag to the NVIDIA OCI hook

Signed-off-by: Jean Rouge <jer329@cornell.edu>
---
 contrib/nvidia/nvidia.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/contrib/nvidia/nvidia.go b/contrib/nvidia/nvidia.go
index e82a51e39..5a1d3c7f1 100644
--- a/contrib/nvidia/nvidia.go
+++ b/contrib/nvidia/nvidia.go
@@ -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(),
 		})