一、使用 Cython 打包 Python 代码

1、安装Cython

pip install cython

2、编写python代码

假设你有一个 example.py文件:

# example.pyx
def say_hello(name):
    print(f"Hello, {name}!")

3、创建 setup.py 文件:

# setup.py
from setuptools import setup
from Cython.Build import cythonize

setup(
    ext_modules=cythonize("example.pyx")
)

4、编译 .py 文件:

运行以下命令以生成 .so 文件:

python setup.py build_ext --inplace

这会生成一个名为 example.cpython-<version>-<platform>.so 的文件。

二、使用 C++ 和 Python API

如果你需要更复杂的打包方式,涉及到 C++ 和 Python API,步骤会更复杂,通常包括创建 C++ 代码和使用 pybind11 等工具来包装 Python 接口。

1、安装pybind11

pip install pybind11

2、编写C++代码

// example.cpp
#include <pybind11/pybind11.h>

void say_hello(const std::string &name) {
    std::cout << "Hello, " << name << "!" << std::endl;
}

PYBIND11_MODULE(example, m) {
    m.def("say_hello", &say_hello);
}

3、创建 CMakeLists.txt 文件:

cmake_minimum_required(VERSION 3.0)
project(example)

find_package(pybind11 REQUIRED)

pybind11_add_module(example example.cpp)

4、编译 C++ 代码:

mkdir build
cd build
cmake ..
make

这将生成一个 example.so 文件。

这样,你可以在 Python 中通过 import example 来使用这个 .so 文件。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐