一、下载pybind11
https://github.com/pybind/pybind11
二、vs2019 新建一个空项目
1.修改输出文件格式 和 文件名称(需要和模块名称一样这个很关键)
只有一个main.cpp函数
#include "pybind11/pybind11.h"
int add(int i, int j)
{
return i + j;
}
PYBIND11_MODULE(example , m) // 模块名example
{
m.doc() = "add plugin";
m.def("add", &add, "add function (i, j)"); // 函数名,函数地址,描述
}
2.配置相关头文件
具体操作步骤可以查看vs2013引用第三方动态链接库,设置include,lib,dll路径问题_cbzhunian的博客-CSDN博客
配置pybind11 的头文件、配置python环境的头文件
(最好是新建一个新的虚拟环境,配置虚拟环境查看conda 安装 Pytorch(GPU)_cbzhunian的博客-CSDN博客_conda安装pytorch gpu)
3.引入Python库(虚拟环境中的)
4.生成文件(x64版本)
example.pyd
三、使用这个模块
1.将example.pyd文件复制到虚拟环境下的Lib文件夹下
(test是我自己的虚拟环境)
2.使用该模块
import example
print(example.add(1, 2))