onnxruntime源码编译

onnxruntime源码编译

ONNX Runtime 是一个开源的深度学习框架,可以在各种硬件和操作系统平台上运行。下面是 ONNX Runtime 源码的编译步骤。

  1. 安装 Git
    在开始之前,需要先安装 Git。
sudo apt-get update
sudo apt-get install git
  1. 拉取源码。
    进入到项目所需目录,执行命令:
cd onnxruntime
  1. 安装依赖
    需要安装 C++ 编译器、CMake、Python3、Python3-dev。
    还需要安装其他依赖库:

  2. 编译。
    在项目根目录下创建 build 目录,并进入:

mkdir build && cd build

执行 CMake:

cmake ../ -DCMAKE_BUILD_TYPE=Release

这里将 CMAKE_BUILD_TYPE 设置为 release,以获得更好的性能和速度。

然后,执行 make 命令:

make -j4

这里使用了 -j4,指定四个线程编译。
编译完成之后,可以在 build 目录下找到编译好的 ONNX Runtime 可执行文件。

  1. 安装 Python 包。
    执行以下命令安装 Python 包:
cd ../../ && python3 setup.py install

这将会在系统中安装 ONNX Runtime 的 Python 包。
至此 ONNX Runtime 源码的编译就完成了。

posted @ 2023-06-11 08:19  汉学  阅读(1891)  评论(0)    收藏  举报