通过CMake使用。
首先做好CMake编译准备工作
建立一个新的文件夹,放入所有c++代码相关的.h和.cpp文件,并建立CMakeLists.txt,内容如下:
cmake_minimum_required(VERSION 3.14)
project(DRO_pyd)
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
# 设置 pybind11 路径
set(pybind11_DIR "/root/miniconda3/lib/python3.12/site-packages/pybind11/share/cmake/pybind11")
# 明确使用 conda 中的 Python 解释器(Python 3.12)
set(Python3_EXECUTABLE "/root/miniconda3/bin/python")
# 找 Python 和 pybind11
find_package(Python3 REQUIRED COMPONENTS Interpreter Development)
find_package(pybind11 REQUIRED)
# 构建 Python 模块(模块名将决定 import 时的名字)
pybind11_add_module(DRO_pyd main.cpp info.cpp labeling.cpp node.cpp)
通过CMake编译生成.so文件
cd your_project_root
rm -rf build
mkdir build
cd build
cmake ..
make -j$(nproc)
将生成的.so放入python的site-packages中
build目录下可以找到生成的.so文件,把它复制到python中即可,我的目录为/root/miniconda3/lib/python3.12/site-packages/