alteur/nix-dotfiles
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

removed

Browse source
This commit is contained in:
alsaiduq-lab 2025-03-22 18:22:20 -06:00
parent fa22a0d6d7
commit d9d22b7ff7
1 changed files with 0 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

28
modules/torch.nix
View file

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