virchau13/automatic1111-webui-nix
1
1
Fork 0
mirror of https://github.com/virchau13/automatic1111-webui-nix.git synced 2024-11-22 14:14:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

Better environment switching for CUDA

Browse Source
This commit is contained in:
Adam Washington 2023-04-04 11:00:18 +01:00
parent cae02d752c
commit 8e26ccbdb4
No known key found for this signature in database
GPG Key ID: 0D2B93AB0C87BAF1
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
impl.nix
View File

@ -1,7 +1,6 @@
{ pkgs, isCUDA ? true, ... }:
let
cuda_path = if isCUDA then pkgs.cudatoolkit else "";
hardware_deps = with pkgs;
if isCUDA then [
cudatoolkit
@ -36,6 +35,6 @@ pkgs.mkShell rec {
glib
];
LD_LIBRARY_PATH=pkgs.lib.makeLibraryPath buildInputs;
CUDA_PATH=pkgs.cudatoolkit;
EXTRA_LDFLAGS="-L${pkgs.linuxPackages.nvidia_x11}/lib";
CUDA_PATH = pkgs.lib.optionalString isCUDA pkgs.cudatoolkit;
EXTRA_LDFLAGS = pkgs.lib.optionalString isCUDA "-L${pkgs.linuxPackages.nvidia_x11}/lib";
}