gpu-issues/flake.nix at main · bubblepipe/gpu-issues · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
{
  description = "GPU bugs analysis environment";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-24.11";
    flake-utils.url = "github:numtide/flake-utils";
  };

  outputs = { self, nixpkgs, flake-utils }:
    flake-utils.lib.eachDefaultSystem (system:
      let
        pkgs = import nixpkgs {
          inherit system;
        };

        pythonPackages = pkgs.python311Packages;
        pythonWithPackages = pythonPackages.python.withPackages (ps: with ps; [
          json5
          matplotlib
          numpy
          openai
          pandas
          requests
          result
          scipy
          seaborn
          scikit-learn
        ]);
      in
      {
        devShells.default = pkgs.mkShell {
          buildInputs = with pkgs; [
            pythonWithPackages
          ];

          shellHook = ''
            echo "Python environment ready with analysis packages"
          '';
        };
      });
}