49 lines
860 B
Nix
49 lines
860 B
Nix
{
|
|
config,
|
|
pkgs,
|
|
lib,
|
|
...
|
|
}:
|
|
let
|
|
customPkgs = import ../pkgs { inherit pkgs lib; };
|
|
in {
|
|
options.custom.python = {
|
|
enable = lib.mkEnableOption "System Python Env";
|
|
};
|
|
|
|
config = lib.mkIf config.custom.python.enable {
|
|
environment.systemPackages = with pkgs; [
|
|
(python311.withPackages (ps: with ps; [
|
|
virtualenv
|
|
(torch.override { cudaSupport = true; })
|
|
torchvision
|
|
torchaudio
|
|
ipython
|
|
i3ipc
|
|
xlib
|
|
six
|
|
psutil
|
|
pynvml
|
|
pyqtgraph
|
|
pyqt6
|
|
numpy
|
|
pandas
|
|
matplotlib
|
|
scipy
|
|
requests
|
|
click
|
|
typer
|
|
rich
|
|
pyyaml
|
|
pytz
|
|
onnxruntime
|
|
pillow
|
|
timm
|
|
] ++ [ customPkgs.python-rembg ]))
|
|
python3Packages.pip
|
|
uv
|
|
ruff
|
|
black
|
|
];
|
|
};
|
|
}
|