Improve obs slightly

This commit is contained in:
Torjus Håkestad 2024-07-16 01:40:36 +02:00
parent 2bece06f10
commit 4c2d93eae7

View File

@ -1,10 +1,11 @@
{ 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
(old: {
# TODO: Remove when fixed in nixpkgs # TODO: Remove when fixed in nixpkgs
# https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154 # https://github.com/NixOS/nixpkgs/blob/nixos-unstable/pkgs/development/libraries/onnxruntime/default.nix#L154
buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ]; buildInputs = old.buildInputs ++ [ pkgs.cudaPackages.nccl ];