开始使用 ONNX Runtime Node.js 绑定

目录

• 安装
• 导入
• 示例
• 支持的版本

安装

# install latest release version
npm install onnxruntime-node

导入

// use ES6 style import syntax (recommended)
import * as ort from 'onnxruntime-node';

// or use CommonJS style import syntax
const ort = require('onnxruntime-node');

示例

• 请按照 快速入门 说明进行 ONNX Runtime Node.js 绑定。

支持的版本

下表列出了预构建二进制文件提供的 ONNX Runtime Node.js 绑定的支持版本。

EPs/平台	Windows x64	Windows arm64	Linux x64	Linux arm64	MacOS x64	MacOS arm64
CPU	✔️	✔️	✔️	✔️	✔️	✔️
DirectML	✔️	✔️	❌	❌	❌	❌
CUDA	❌	❌	✔️[1]	❌	❌	❌

• [1]: CUDA v11.8。

对于未在列表中的平台或需要自定义构建，您可以从源代码构建 Node.js 绑定，并使用 npm install <onnxruntime_repo_root>/js/node/ 进行使用。