python 通过 pybind11 使用Eigen加速代码的步骤
作者:好问雷 时间:2021-03-07 06:09:30
python是很容易上手的编程语言,但是有些时候使用python编写的程序并不能保证其运行速度(例如:while 和 for),这个时候我们就需要借助c++等为我们的代码提速。下面是我使用pybind11调用c++的Eigen库的简单介绍:
第一步:准备系统和IDE:
Windows 10
vs2015 (用于调试c++代码)
vscode (调试python代码)
第二步:python虚拟环境:
1.创建虚拟python虚拟环境: 在vscode的terminal中执行
python -m venv env
2.下载 Eigen: 将Eigen解压到当前目录命名为 eigen-3.3.8
3.在vscode的terminal中激活虚拟环境:
./env/Scripts/Activate.ps1
4.安装pybind11:
pip install pybind11
5.安装numpy==1.19.3(使用1.19.4可能会有问题)
pip install numpy==1.19.3
第三步:使用vs2015编写cpp_python.cpp, 并保证没有bug
#include <Eigen/Dense>
using namespace std
using namespace Eigen
MatrixXd add_mat(MatrixXd A_mat, MatrixXd B_mat)
{
return A_mat + B_mat;
}
第四步:使用pybind11为cpp_python.cpp添加python接口
// cpp_python.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include <pybind11/pybind11.h>
#include <pybind11/eigen.h>
#include<pybind11/numpy.h>
#include<fstream>
#include<iostream>
#include <Eigen/Dense>
using namespace std;
using namespace Eigen;
MatrixXd add_mat(MatrixXd A_mat, MatrixXd B_mat)
{
return A_mat + B_mat;
}
namespace py = pybind11;
PYBIND11_MODULE(add_mat_moudle, m)
{
m.doc() = "Matrix add";//解释说明
m.def("mat_add_py"/*在pyhon中使用的函数名*/, &add_mat);
}
第五步:设置setup.py用来编译c++代码
from setuptools import setup
from setuptools import Extension
add_mat_module = Extension(name='add_mat_moudle', # 模块名称
sources=['cpp_python.cpp'], # 源码
include_dirs=[r'.\eigen-3.3.8',
r'.\env\Scripts', # 依赖的第三方库的头文件
r'.\env\Lib\site-packages\pybind11\include']
)
setup(ext_modules=[add_mat_module])
第六步:编译测试
注意:我的cpp_python.cpp和setup.py是在同一个文件夹下。
执行: "python .\setup.py build_ext --inplace"就会得下面的结果,生成.pyd文件表明我们已经编译成功。
运行测试:
来源:https://www.cnblogs.com/hyl2018/p/14089299.html
标签:python,加速代码,pybind11,Eigen
0
投稿
猜你喜欢
pandas 实现某一列分组,其他列合并成list
2022-06-12 13:49:21
python使用PIL模块获取图片像素点的方法
2022-07-28 10:57:57
Python多线程编程(八):使用Event实现线程间通信
2023-05-07 15:25:50
JavaScript中实现字符串的取左取右(实现left和right功能)
2009-07-20 12:39:00
Python破解excel进入密码的过程详解
2021-06-14 03:06:25
Python求两点之间的直线距离(2种实现方法)
2021-03-28 19:24:46
Django-silk性能测试工具安装及使用解析
2023-12-08 08:53:19
pandas学习之df.set_index的具体使用
2021-11-12 07:35:20
PHP PDOStatement::bindColumn讲解
2023-06-10 04:35:55
php抓取页面的几种方法详解
2023-11-14 10:53:42
基于PyTorch实现一个简单的CNN图像分类器
2021-08-16 21:52:31
python snownlp情感分析简易demo(分享)
2021-07-18 04:32:35
举例讲解Django中数据模型访问外键值的方法
2022-05-29 18:40:11
Python中输入若干整数以逗号间隔实现统计每个整数出现次数
2021-10-27 22:20:04
Refactoring HTML 书评
2008-07-10 12:00:00
Python SDK实现私服上传下载的示例
2021-11-22 08:14:42
Python实现动态添加类的属性或成员函数的解决方法
2021-05-13 17:13:41
Python中的Decorator装饰器的使用示例
2023-07-02 10:37:44
python中itertools模块zip_longest函数详解
2023-01-02 09:09:35
python实现邮件自动发送
2023-06-10 16:24:44