Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .envrc
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
use flake
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ user_path_config-deprecated.txt
/.coverage*
/auth.json
.DS_Store
.direnv
61 changes: 61 additions & 0 deletions flake.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

62 changes: 62 additions & 0 deletions flake.nix
Original file line number Diff line number Diff line change
@@ -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
'';
};
});
}