added cuda support
This commit is contained in:
1 parent
fb816cdf83
commit
8ddda784a4
1 file changed
+7
-3
@@ -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: [ ])) ];
|
||||
};
|
||||
});
|
||||
}
|
||||
Reference in new issue
Block a user