From 8ddda784a4be2791e30a84c60ef13f5ef471096f Mon Sep 17 00:00:00 2001
From: didier amyot <didier.amyot@all-dressed-programming.com>
Date: Mon, 28 Oct 2024 05:05:34 -0400
Subject: [PATCH] added cuda support

---
 flake.nix | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/flake.nix b/flake.nix
index ff9dc15..00e178a 100644
--- a/flake.nix
+++ b/flake.nix
@@ -8,7 +8,12 @@
   outputs = { self, nixpkgs, flake-utils }:
     flake-utils.lib.eachDefaultSystem (system:
       let
-        pkgs = nixpkgs.legacyPackages.${system};
+        pkgs = import nixpkgs {
+          system = system;
+          config.allowUnfree = true;
+          config.cudaSupport = true;
+        };
+        # pkgs = nixpkgs.legacyPackages.${system};
 
         pythonPackages = pkgs.python312Packages;
 
@@ -189,8 +194,7 @@
           python312Packages.opencv4 = custom-opencv4;
         };
         devShells.default = pkgs.mkShell {
-          buildInputs =
-            [ (pkgs.python312.withPackages (ps: [ custom-opencv4 ])) ];
+          buildInputs = [ (pkgs.python312.withPackages (ps: [ ])) ];
         };
       });
 }