KCNHUB: Python: Add pycuda, accelerate & datasets

2024-11-04 13:51:22 -05:00 · 2024-11-04 13:51:22 -05:00 · 606ae11bef
parent 5a889d63f0
commit 606ae11bef
1 changed files with 3 additions and 0 deletions
--- a/machines/kcnhub/packages/python.nix
+++ b/machines/kcnhub/packages/python.nix
@ -59,6 +59,9 @@
      torchdiffeq
      botorch
      lion-pytorch
+      pycuda
+      accelerate
+      datasets
    ]);
  };
 }