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

Add workaround for onnxruntime with cuda
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-15 15:25:06 +02:00
parent 499a6a8eb5
commit 2bece06f10
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,9 +1,15 @@
{ pkgs, lib, osConfig, ... }:
let
withCuda = if (osConfig.system.name == "gunter") then true else false;
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 ];
});
obs-backgrounremoval-gpu = ((pkgs.obs-studio-plugins.obs-backgroundremoval.override {
onnxruntime = onnxruntime-gpu;
}).overrideAttrs (old: {