https://github.com/pybind/pybind11


只有一个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)"); // 函数名,函数地址,描述
- }
具体操作步骤可以查看vs2013引用第三方动态链接库,设置include,lib,dll路径问题_cbzhunian的博客-CSDN博客
(最好是新建一个新的虚拟环境,配置虚拟环境查看conda 安装 Pytorch(GPU)_cbzhunian的博客-CSDN博客_conda安装pytorch gpu)


example.pyd
(test是我自己的虚拟环境)

- import example
-
-
- print(example.add(1, 2))
