From 606ae11befb5e3e073112641304d04832a00be2b Mon Sep 17 00:00:00 2001
From: David Crompton <david.crompton@mail.utoronto.ca>
Date: Mon, 4 Nov 2024 13:51:22 -0500
Subject: [PATCH] KCNHUB: Python: Add pycuda, accelerate & datasets

---
 machines/kcnhub/packages/python.nix | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/machines/kcnhub/packages/python.nix b/machines/kcnhub/packages/python.nix
index ebbef73..f271096 100644
--- a/machines/kcnhub/packages/python.nix
+++ b/machines/kcnhub/packages/python.nix
@@ -59,6 +59,9 @@
       torchdiffeq
       botorch
       lion-pytorch
+      pycuda
+      accelerate
+      datasets
     ]);
   };
 }