torjus/nixos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Releases Activity

Improve obs slightly
Some checks are pending
pre-commit / pre-commit (push) Waiting to run
Details

Browse Source
This commit is contained in:
Torjus Håkestad 2024-07-16 01:40:36 +02:00
parent 2bece06f10
commit 4c2d93eae7
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
home/programs/obs-studio/default.nix
View File

@ -1,14 +1,15 @@
{ pkgs, lib, osConfig, ... }:
let
withCuda = if (osConfig.system.name == "gunter") then true else false;
withCuda = osConfig.system.name == "gunter";
onnxruntime-gpu = (pkgs.onnxruntime.override {
cudaSupport = withCuda;
}).overrideAttrs (old: {
# TODO: Remove when fixed in nixpkgs
# https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154
buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ];
});
}).overrideAttrs
(old: {
# TODO: Remove when fixed in nixpkgs
# 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 {
onnxruntime = onnxruntime-gpu;