nixos-config/pkgs/onnx-runtime/default.nix

{ stdenv }:
stdenv.mkDerivation {
  pname = "onnx-runtime";
  version = "1.14.1";
  # phases = [ "unpackPhase" "installPhase" ];

  src = builtins.fetchTarball {
    url = "https://github.com/microsoft/onnxruntime/releases/download/v1.14.1/onnxruntime-linux-x64-gpu-1.14.1.tgz";
    sha256 = "1hcxyhldxy5q0x50f96ni3p9v1n5qljarx7fvs63r6sk6j4yr2kv";
  };

  installPhase = ''
    mkdir -p $out/include/onnxruntime/core/session
    mkdir -p $out/include/onnxruntime/core/providers/cpu
    mkdir -p $out/include/onnxruntime/core/providers/gpu

    cp -r lib $out/
    cp -r include/ $out/
  '';
}