20 lines
626 B
Nix
20 lines
626 B
Nix
|
{
|
||
|
|
inputs.nixpkgsUnstable.url = "github:nixos/nixpkgs";
|
||
|
|
|
||
|
|
outputs = { self, nixpkgs, nixpkgsUnstable, flake-utils }:
|
||
|
|
flake-utils.lib.eachDefaultSystem (system:
|
||
|
|
let
|
||
|
|
pkgs = import nixpkgs { inherit system; };
|
||
|
|
pkgsUnstable = import nixpkgsUnstable {
|
||
|
|
inherit system;
|
||
|
|
config.allowUnfreePredicate = pkg:
|
||
|
|
builtins.elem (nixpkgs.lib.getName pkg) [ "cudatoolkit" ];
|
||
|
|
};
|
||
|
|
in {
|
||
|
|
devShell = pkgs.mkShell {
|
||
|
|
packages = (with pkgs; [ clang-tools gdb octave ])
|
||
|
|
++ (with pkgsUnstable.cudaPackages_12; [ cudatoolkit ]);
|
||
|
|
};
|
||
|
|
});
|
||
|
|
}
|