andrey/onnxruntime-rs
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity
Rust bindings to onnxruntime
112 Commits 9 Branches 0 Tags 190 KiB
Rust 89%
Nix 9.2%
Shell 1.4%
C 0.4%
7a6101bee7
Go to file
Andrey Tkachenko 7a6101bee7 make available_providers function
2021-01-22 14:58:21 +04:00
.github Build with release 1.4.0 in CI 2020-08-23 15:00:41 +01:00
cbits Fix bindings for CUDA Provider 2021-01-22 13:45:06 +04:00
examples make available_providers function 2021-01-22 14:58:21 +04:00
nix niv update 2020-07-05 23:41:48 +02:00
src make available_providers function 2021-01-22 14:58:21 +04:00
testdata Add testdata to repo 2020-05-31 09:13:59 +01:00
.gitignore Setup cargo actions 2020-07-05 17:32:38 +02:00
build.rs Fix bindings for CUDA Provider 2021-01-22 13:45:06 +04:00
Cargo.lock Some fixes 2021-01-15 14:19:20 +04:00
Cargo.toml Some fixes 2021-01-15 14:19:20 +04:00
default.nix Expect the C headers in a different location 2020-07-05 23:16:07 +02:00
LICENSE Add MIT license 2020-05-31 09:58:02 +01:00
README.md Update README 2020-08-23 15:33:23 +01:00

README.md

onnxruntime

Cargo CI Nix CI

This crate provides bindings to onnxruntime. Raw bindings to the C API (sys module) as well as some opinionated wrappers to make onnxruntime usable safely from plain Rust.

In order to build and use this crate with cargo you must either have onnxruntime installed on your system or point to the release (or build) of onnxruntime you want to use manually.

Tested (on Linux) with onnxruntime versions:

Download the latest release of onnxruntime

Download from: https://github.com/microsoft/onnxruntime/releases

Then either install it globally (for example /usr/local/{lib,include/) or export the following environment variables:

ONNXRUNTIME_LIB_DIR: path to the lib directory of the release ONNXRUNTIME_INCLUDE_DIR: path to the include directory of the release

Also make sure the lib directory is also part of LD_LIBRARY_PATH.

Build onnxruntime from source

Follow the guide at: https://github.com/microsoft/onnxruntime/blob/master/BUILD.md

For Linux/Mac OS X:

  • The lib directory will be: onnxruntime/build/Linux/RelWithDebInfo/
  • The include directory will be: onnxruntime/include/onnxruntime/core/session/

In onnxruntime directory:

./build.sh --config RelWithDebInfo --build_shared_lib --parallel
export ONNXRUNTIME_LIB_DIR=$PWD/build/Linux/RelWithDebInfo/
export ONNXRUNTIME_INCLUDE_DIR=$PWD/include
export LD_LIBRARY_PATH=$ONNXRUNTIME_LIB_DIR:$LD_LIBRARY_PATH

License

MIT License

Copyright 2020 Basile Henry, Chris Chalmers