77 lines
2.2 KiB
Bash
77 lines
2.2 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
# Download Microsoft's CPU-dispatched ONNX Runtime (not pyke's AVX2-static build).
|
||
|
|
# Usage: fetch-onnxruntime.sh [DEST_DIR]
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
VERSION="${ORT_VERSION:-1.24.1}"
|
||
|
|
DEST="${1:-}"
|
||
|
|
if [ -z "$DEST" ]; then
|
||
|
|
DEST="${DATA_DIR:-./data}/onnxruntime"
|
||
|
|
fi
|
||
|
|
|
||
|
|
os_raw="${TARGETOS:-$(uname -s | tr '[:upper:]' '[:lower:]')}"
|
||
|
|
arch_raw="${TARGETARCH:-$(uname -m)}"
|
||
|
|
|
||
|
|
case "$os_raw" in
|
||
|
|
linux|darwin|macos) ;;
|
||
|
|
*) echo "unsupported OS: $os_raw" >&2; exit 1 ;;
|
||
|
|
esac
|
||
|
|
[ "$os_raw" = macos ] && os_raw=darwin
|
||
|
|
|
||
|
|
case "$arch_raw" in
|
||
|
|
amd64|x86_64|x64) arch_raw=x86_64 ;;
|
||
|
|
arm64|aarch64) arch_raw=aarch64 ;;
|
||
|
|
esac
|
||
|
|
|
||
|
|
if [ "$os_raw" = linux ]; then
|
||
|
|
if [ "$arch_raw" = x86_64 ]; then
|
||
|
|
artifact="onnxruntime-linux-x64-${VERSION}.tgz"
|
||
|
|
sha256="9142552248b735920f9390027e4512a2cacf8946a1ffcbe9071a5c210531026f"
|
||
|
|
elif [ "$arch_raw" = aarch64 ]; then
|
||
|
|
artifact="onnxruntime-linux-aarch64-${VERSION}.tgz"
|
||
|
|
sha256="0f56edd68f7602df790b68b874a46b115add037e88385c6c842bb763b39b9f89"
|
||
|
|
else
|
||
|
|
echo "unsupported Linux arch: $arch_raw" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
dylib_glob="libonnxruntime.so*"
|
||
|
|
elif [ "$os_raw" = darwin ]; then
|
||
|
|
if [ "$arch_raw" = aarch64 ]; then
|
||
|
|
artifact="onnxruntime-osx-arm64-${VERSION}.tgz"
|
||
|
|
else
|
||
|
|
artifact="onnxruntime-osx-universal2-${VERSION}.tgz"
|
||
|
|
fi
|
||
|
|
sha256=""
|
||
|
|
dylib_glob="libonnxruntime.dylib"
|
||
|
|
else
|
||
|
|
echo "unsupported OS: $os_raw" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
url="https://github.com/microsoft/onnxruntime/releases/download/v${VERSION}/${artifact}"
|
||
|
|
mkdir -p "$DEST"
|
||
|
|
stamp="$DEST/.ort-${VERSION}-$(basename "$artifact" .tgz)"
|
||
|
|
if [ -f "$stamp" ]; then
|
||
|
|
dylib=$(find "$DEST" -maxdepth 3 \( -name 'libonnxruntime.so.1.*' -o -name 'libonnxruntime.dylib' \) | head -n 1)
|
||
|
|
if [ -n "$dylib" ]; then
|
||
|
|
printf '%s\n' "$dylib"
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
tmp=$(mktemp)
|
||
|
|
trap 'rm -f "$tmp"' EXIT
|
||
|
|
echo "downloading ONNX Runtime ${VERSION} ($artifact)" >&2
|
||
|
|
curl -fsSL "$url" -o "$tmp"
|
||
|
|
if [ -n "$sha256" ]; then
|
||
|
|
echo "$sha256 $tmp" | sha256sum -c - >&2
|
||
|
|
fi
|
||
|
|
tar -xzf "$tmp" -C "$DEST" --strip-components=1
|
||
|
|
touch "$stamp"
|
||
|
|
dylib=$(find "$DEST" -maxdepth 3 \( -name 'libonnxruntime.so.1.*' -o -name 'libonnxruntime.dylib' \) | head -n 1)
|
||
|
|
if [ -z "$dylib" ]; then
|
||
|
|
echo "ONNX Runtime dylib missing after extract in $DEST" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
printf '%s\n' "$dylib"
|