欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python 通过 pybind11 使用Eigen加速代码的步骤

程序员文章站 2022-07-03 22:07:06
python是很容易上手的编程语言,但是有些时候使用python编写的程序并不能保证其运行速度(例如:while 和 for),这个时候我们就需要借助c++等为我们的代码提速。下面是我使用pybind...

python是很容易上手的编程语言,但是有些时候使用python编写的程序并不能保证其运行速度(例如:while 和 for),这个时候我们就需要借助c++等为我们的代码提速。下面是我使用pybind11调用c++的eigen库的简单介绍:

第一步:准备系统和ide:

windows 10 
vs2015 (用于调试c++代码)
vscode (调试python代码)

第二步:python虚拟环境:

1.创建虚拟python虚拟环境: 在vscode的terminal中执行 

2.下载 eigen: 将eigen解压到当前目录命名为 eigen-3.3.8
3.在vscode的terminal中激活虚拟环境:

 ./env/scripts/activate.ps1

python 通过 pybind11 使用Eigen加速代码的步骤

4.安装pybind11:

5.安装numpy==1.19.3(使用1.19.4可能会有问题)

第三步:使用vs2015编写cpp_python.cpp, 并保证没有bug

第四步:使用pybind11为cpp_python.cpp添加python接口

第五步:设置setup.py用来编译c++代码

第六步:编译测试

python 通过 pybind11 使用Eigen加速代码的步骤

注意:我的cpp_python.cpp和setup.py是在同一个文件夹下。

执行: "python .\setup.py build_ext --inplace"就会得下面的结果,生成.pyd文件表明我们已经编译成功。

python 通过 pybind11 使用Eigen加速代码的步骤

运行测试:

python 通过 pybind11 使用Eigen加速代码的步骤

以上就是python 通过 pybind11 使用eigen加速代码的步骤的详细内容,更多关于python 加速代码的资料请关注其它相关文章!