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 ]); }; }