diff --git a/flake.nix b/flake.nix
index f73df9c..e368cb4 100644
--- a/flake.nix
+++ b/flake.nix
@@ -39,6 +39,7 @@
               lib.flatten [
                 (cachix "nix-community")
                 "https://cache.nixos.org"
+                "https://cache.nixos-cuda.org"
               ];
           };
         };
diff --git a/nixos/modules/nix.nix b/nixos/modules/nix.nix
index b80b30f..becc392 100644
--- a/nixos/modules/nix.nix
+++ b/nixos/modules/nix.nix
@@ -28,10 +28,6 @@
           ];
         packageOverrides = p: {
           intel-vaapi-driver = p.intel-vaapi-driver.override {enableHybridCodec = true;};
-          onnxruntime = p.onnxruntime.override {
-            cudaSupport = false;
-            ncclSupport = false;
-          };
         };
       };
     };
@@ -51,6 +47,7 @@
         ];
         trusted-public-keys = [
           "nix-community.cachix.org-1:mB9FSh9qf2dCimDSUo8Zy7bkq5CX+/rkCWyvRCYg3Fs="
+          "cache.nixos-cuda.org:74DUi4Ye579gUqzH4ziL9IyiJBlDpMRn9MBN8oNan9M="
         ];
         auto-optimise-store = lib.mkDefault true;
         experimental-features = [