27 lines
428 B
Nix
27 lines
428 B
Nix
{ pkgs, ... }:
|
|
let
|
|
native_dependencies = with pkgs.python312Packages; [
|
|
torch-bin
|
|
torchvision-bin
|
|
aggdraw # for visualtorch
|
|
nvidia-ml-py
|
|
];
|
|
tools = with pkgs; [
|
|
ruff
|
|
dmidecode
|
|
];
|
|
in
|
|
{
|
|
packages = native_dependencies ++ tools;
|
|
languages.python = {
|
|
enable = true;
|
|
package = pkgs.python312;
|
|
uv = {
|
|
enable = true;
|
|
sync.enable = true;
|
|
};
|
|
venv.enable = true;
|
|
};
|
|
|
|
}
|