This commit is contained in:
parent
2bece06f10
commit
4c2d93eae7
@ -1,10 +1,11 @@
|
||||
{ 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: {
|
||||
}).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 ];
|
||||
|
Loading…
Reference in New Issue
Block a user