diff --git a/modules/torch.nix b/modules/torch.nix deleted file mode 100644 index 0c8aa58..0000000 --- a/modules/torch.nix +++ /dev/null @@ -1,28 +0,0 @@ -{ config, pkgs, lib, ... }: - -with lib; - -let - cfg = config.custom.torch; -in { - options.custom.torch = { - enable = mkEnableOption "PyTorch with CUDA support"; - cudaSupport = mkOption { - type = types.bool; - default = true; - }; - package = mkOption { - type = types.package; - default = pkgs.python311Packages.torch.override { cudaSupport = cfg.cudaSupport; }; - }; - }; - - config = mkIf cfg.enable { - environment.systemPackages = with pkgs; [ - (python3.withPackages (ps: with ps; [ - cfg.package - numpy - ])) - ]; - }; -}