summaryrefslogtreecommitdiff
path: root/cuda/python/devenv.nix
blob: c32e55fe1ac3e3280ac701108cf149b01d925112 (plain)
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
{
  pkgs,
  lib,
  config,
  inputs,
  ...
}: let
  cpkgs = import inputs.nixpkgs {
    system = pkgs.system;
    config = {
      allowUnfree = true;
      cudaSupport = true;
    };
  };
in {
  # https://devenv.sh/basics/
  env.GREET = "devenv";

  # env.LD_LIBRARY_PATH = "/run/opengl-driver/lib";
  # env.LD_LIBRARY_PATH = "${pkgs.glibc}/lib";
  # env.CUDA_HOME = pkgs.cudaPackages.cudatoolkit;
  # env.CUDA_PATH = pkgs.cudaPackages.cudatoolkit;

  # env.LD_LIBRARY_PATH = "/run/opengl-driver/lib:" + lib.makeLibraryPath config.packages;
  # env.LD_LIBRARY_PATH = "/run/opengl-driver/lib";
  # # # Ensure Nix packages are discoverable by Python
  # env.PYTHONPATH = lib.makeSearchPath "lib/python3.13/site-packages" pythonPkgs;

  # https://devenv.sh/packages/
  packages = with cpkgs; [
    git
    # glib
    # glibc
    # libGL
    # libGLU
    # cudaPackages.cudatoolkit
    # cudaPackages.cuda_cccl
    # cudaPackages.cudnn
    # cudaPackages.cuda_nvcc
    # stdenv.cc.cc.lib
    # zlib
    python3.withPackages
    (ps:
      with ps; [
        torch
      ])
  ];

  # https://devenv.sh/languages/
  # languages.rust.enable = true;

  languages.python = {
    enable = true;
    venv.enable = true;
  };
  # https://devenv.sh/processes/
  # processes.dev.exec = "${lib.getExe pkgs.watchexec} -n -- ls -la";

  # https://devenv.sh/services/
  # services.postgres.enable = true;

  # https://devenv.sh/scripts/
  scripts.hello.exec = ''
    echo hello from $GREET
  '';

  # https://devenv.sh/basics/
  enterShell = ''
    hello         # Run scripts directly
    git --version # Use packages
  '';

  # https://devenv.sh/tasks/
  # tasks = {
  #   "myproj:setup".exec = "mytool build";
  #   "devenv:enterShell".after = [ "myproj:setup" ];
  # };

  # https://devenv.sh/tests/
  enterTest = ''
    echo "Running tests"
    git --version | grep --color=auto "${pkgs.git.version}"
  '';

  # https://devenv.sh/git-hooks/
  # git-hooks.hooks.shellcheck.enable = true;

  # See full reference at https://devenv.sh/reference/options/
}