parent
2bece06f10
commit
4c2d93eae7
@ -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 ];
|
||||||
|
Loading…
Reference in New Issue
Block a user