Add the nvidia runtime cdi

Signed-off-by: manuelbuil <[email protected]>
manuelbuil · Oct 11, 2024 · eff8246 · eff8246
1 parent 46cfd2c
commit eff8246
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/agent/containerd/runtimes.go b/pkg/agent/containerd/runtimes.go
@@ -71,6 +71,10 @@ func findNvidiaContainerRuntimes(foundRuntimes runtimeConfigs) {
 			RuntimeType: "io.containerd.runc.v2",
 			BinaryName:  "nvidia-container-runtime-experimental",
 		},
+                "nvidia-cdi": {
+                        RuntimeType: "io.containerd.runc.v2",
+                        BinaryName:  "nvidia-container-runtime.cdi",
+                },
 	}
 
 	searchForRuntimes(potentialRuntimes, foundRuntimes)