20 lines
593 B
Nix
20 lines
593 B
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/
|
|
'';
|
|
}
|