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 }: outputs = { self, nixpkgs, flake-utils }:
flake-utils.lib.eachDefaultSystem (system: flake-utils.lib.eachDefaultSystem (system:
let let
pkgs = nixpkgs.legacyPackages.${system}; pkgs = import nixpkgs {
system = system;
config.allowUnfree = true;
config.cudaSupport = true;
};
# pkgs = nixpkgs.legacyPackages.${system};
pythonPackages = pkgs.python312Packages; pythonPackages = pkgs.python312Packages;
@@ -189,8 +194,7 @@
python312Packages.opencv4 = custom-opencv4; python312Packages.opencv4 = custom-opencv4;
}; };
devShells.default = pkgs.mkShell { devShells.default = pkgs.mkShell {
buildInputs = buildInputs = [ (pkgs.python312.withPackages (ps: [ ])) ];
[ (pkgs.python312.withPackages (ps: [ custom-opencv4 ])) ];
}; };
}); });
} }