28 lines
432 B
Nix
28 lines
432 B
Nix
|
{ pkgs, ... }:
|
||
|
|
let
|
||
|
|
native_dependencies = with pkgs.python312Packages; [
|
||
|
|
torch-bin
|
||
|
|
torchvision-bin
|
||
|
|
aggdraw # for visualtorch
|
||
|
|
numpy
|
||
|
|
scipy
|
||
|
|
matplotlib
|
||
|
|
];
|
||
|
|
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;
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|