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