diff --git a/nix/flake.nix b/nix/flake.nix
index 40883ac..2865925 100644
--- a/nix/flake.nix
+++ b/nix/flake.nix
@@ -91,6 +91,7 @@
           stateVersion = "25.05";
           nixpkgsConfig = {
             rocmSupport = true;
+            cudaSupport = true;
           };
         })
         (mkHost {