feat: add flake for nix users

This commit is contained in:
Arthurdw 2025-04-01 14:08:50 +02:00
parent 59f183ab9b
commit c1f2baca5e
No known key found for this signature in database
GPG Key ID: 30393DC036C318EF
4 changed files with 125 additions and 0 deletions

1
.envrc Normal file
View File

@ -0,0 +1 @@
use flake

1
.gitignore vendored
View File

@ -53,3 +53,4 @@ user_path_config-deprecated.txt
/.coverage* /.coverage*
/auth.json /auth.json
.DS_Store .DS_Store
.direnv

61
flake.lock Normal file
View File

@ -0,0 +1,61 @@
{
"nodes": {
"flake-utils": {
"inputs": {
"systems": "systems"
},
"locked": {
"lastModified": 1731533236,
"narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
"owner": "numtide",
"repo": "flake-utils",
"rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
"type": "github"
},
"original": {
"owner": "numtide",
"repo": "flake-utils",
"type": "github"
}
},
"nixpkgs": {
"locked": {
"lastModified": 1742069588,
"narHash": "sha256-C7jVfohcGzdZRF6DO+ybyG/sqpo1h6bZi9T56sxLy+k=",
"owner": "nixos",
"repo": "nixpkgs",
"rev": "c80f6a7e10b39afcc1894e02ef785b1ad0b0d7e5",
"type": "github"
},
"original": {
"owner": "nixos",
"ref": "nixos-unstable",
"repo": "nixpkgs",
"type": "github"
}
},
"root": {
"inputs": {
"flake-utils": "flake-utils",
"nixpkgs": "nixpkgs"
}
},
"systems": {
"locked": {
"lastModified": 1681028828,
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
"owner": "nix-systems",
"repo": "default",
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
"type": "github"
},
"original": {
"owner": "nix-systems",
"repo": "default",
"type": "github"
}
}
},
"root": "root",
"version": 7
}

62
flake.nix Normal file
View File

@ -0,0 +1,62 @@
{
inputs = {
nixpkgs.url = "github:nixos/nixpkgs?ref=nixos-unstable";
flake-utils.url = "github:numtide/flake-utils";
};
outputs = { self, nixpkgs, flake-utils }:
flake-utils.lib.eachDefaultSystem (system:
let
pkgs = import nixpkgs {
inherit system;
config.allowUnfree = true;
};
venv-dir = "fooocus_env";
in {
devShells.default = pkgs.mkShell {
buildInputs = with pkgs; [
# For more packages/package search go to https://search.nixos.org/
python310 # Adjust Python version as needed
python310Packages.pip
python310Packages.virtualenv
pkg-config
stdenv.cc.cc.lib
cudatoolkit
cudaPackages.cudnn
cudaPackages.cuda_cudart
gcc13
zlib
];
shellHook = ''
export CUDA_PATH=${pkgs.cudatoolkit}
export CC=${pkgs.gcc13}/bin/gcc
export CXX=${pkgs.gcc13}/bin/g++
export PATH=${pkgs.gcc13}/bin:$PATH
export LD_LIBRARY_PATH=${
pkgs.lib.makeLibraryPath [
"/run/opengl-driver"
pkgs.cudatoolkit
pkgs.cudaPackages.cudnn
pkgs.stdenv.cc.cc
]
}:$LD_LIBRARY_PATH
export LIBRARY_PATH=${
pkgs.lib.makeLibraryPath [ pkgs.cudatoolkit ]
}:$LIBRARY_PATH
if [ ! -d ${venv-dir} ]; then
echo "Creating virtual environment..."
python3 -m venv ${venv-dir}
fi
echo "Activating virtual environment..."
source ${venv-dir}/bin/activate
pip install -r ./requirements_versions.txt
'';
};
});
}