This commit is contained in:
Jan Kowalczyk
2025-08-13 14:16:14 +02:00
parent a936b754cb
commit cd4dc583e8
2 changed files with 41 additions and 0 deletions

View File

@@ -0,0 +1,24 @@
{ pkgs, ... }:
let
native_dependencies = with pkgs.python312Packages; [
torch-bin
torchvision-bin
aggdraw # for visualtorch
];
tools = with pkgs; [
ruff
];
in
{
packages = native_dependencies ++ tools;
languages.python = {
enable = true;
package = pkgs.python312;
uv = {
enable = true;
sync.enable = true;
};
venv.enable = true;
};
}