feat: neuro: some

2026-02-27 12:53:51 +00:00
parent 594ca4156f
commit fa31b8ac06
1 changed files with 16 additions and 6 deletions
@@ -113,12 +113,22 @@
  zramSwap.swapDevices   = 1;
  zramSwap.memoryPercent = lib.mkDefault 100;

-  environment.systemPackages = with pkgs; [
-    # (pkgs.python3.withPackages (ps: with ps; [
-    #   hectic.py3-openai-shap-e
-    #   torchWithCuda torchvision pytorch3d
-    #   fvcore iopath tqdm
-    # ]))
+  environment.systemPackages = with pkgs; let
+    python-ai = python3.withPackages (ps: let
+      torchCuda     = ps.torchWithCuda;
+      torchvision   = ps.torchvision.override { torch = torchCuda; };
+      pytorch3dCuda = ps.pytorch3d.override { torch = torchCuda; };
+    in [
+      torchCuda
+      torchvision
+      pytorch3dCuda
+      ps.fvcore
+      ps.iopath
+      ps.tqdm
+      hectic.py3-openai-shap-e  # Uncomment when needed; depends on torch
+    ]);
+  in [
+    python-ai
    git
    neovim
    wget