From bf15308d1e0200bd1fd7de25fbce9709188e4354 Mon Sep 17 00:00:00 2001
From: Michael Thomson <michael@michaelthomson.dev>
Date: Wed, 3 Dec 2025 11:47:42 -0500
Subject: [PATCH] fix: use generic cdi plugin

---
 modules/nvidia.nix | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/modules/nvidia.nix b/modules/nvidia.nix
index 093bb7c..9ec7931 100644
--- a/modules/nvidia.nix
+++ b/modules/nvidia.nix
@@ -22,6 +22,7 @@
     nvidia-container-toolkit
   ];
 
+  services.k3s.nodeLabel = ["nixos-nvidia-cdi=enabled"];
   services.k3s.containerdConfigTemplate = ''
     {{ template "base" . }}
 
@@ -30,8 +31,5 @@
       runtime_engine = ""
       runtime_root = ""
       runtime_type = "io.containerd.runc.v2"
-
-    [plugins."io.containerd.grpc.v1.cri".containerd.runtimes.nvidia.options]
-      BinaryName = "${lib.getOutput "tools" config.hardware.nvidia-container-toolkit.package}/bin/nvidia-container-runtime"
   '';
 }