This commit is contained in:
		| @@ -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 ]; | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user