This commit is contained in:
		| @@ -1,14 +1,15 @@ | |||||||
| { pkgs, lib, osConfig, ... }: | { pkgs, lib, osConfig, ... }: | ||||||
| let | let | ||||||
|   withCuda = if (osConfig.system.name == "gunter") then true else false; |   withCuda = osConfig.system.name == "gunter"; | ||||||
|  |  | ||||||
|   onnxruntime-gpu = (pkgs.onnxruntime.override { |   onnxruntime-gpu = (pkgs.onnxruntime.override { | ||||||
|     cudaSupport = withCuda; |     cudaSupport = withCuda; | ||||||
|   }).overrideAttrs (old: { |   }).overrideAttrs | ||||||
|     # TODO: Remove when fixed in nixpkgs |     (old: { | ||||||
|     # https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154 |       # TODO: Remove when fixed in nixpkgs | ||||||
|     buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ]; |       # https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154 | ||||||
|   }); |       buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ]; | ||||||
|  |     }); | ||||||
|  |  | ||||||
|   obs-backgrounremoval-gpu = ((pkgs.obs-studio-plugins.obs-backgroundremoval.override { |   obs-backgrounremoval-gpu = ((pkgs.obs-studio-plugins.obs-backgroundremoval.override { | ||||||
|     onnxruntime = onnxruntime-gpu; |     onnxruntime = onnxruntime-gpu; | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user