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 }:
|
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 ])) ];
|
|
||||||
};
|
};
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
Reference in new issue
Block a user