added cuda support

This commit is contained in:
didier committed 2024-10-28 05:05:34 -04:00
1 parent fb816cdf83
commit 8ddda784a4
1 file changed
+7 -3
+7 -3
View File
@@ -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: [ ])) ];
};
});
}