python3 通過 pybind11 使用Eigen加速代碼的步驟詳解
更新時間：2020年12月07日 10:21:26 作者：好問雷
這篇文章主要介紹了python3 通過 pybind11 使用Eigen加速代碼的步驟詳解,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
python是很容易上手的編程語言，但是有些時候使用python編寫的程序并不能保證其運行速度（例如：while 和 for），這個時候我們就需要借助c++等為我們的代碼提速。下面是我使用pybind11調用c++的Eigen庫的簡單介紹:
第一步：準備系統(tǒng)和IDE：
- Windows 10
- vs2015 (用于調試c++代碼)
- vscode (調試python代碼)
第二步：python虛擬環(huán)境：
1.創(chuàng)建虛擬python虛擬環(huán)境： 在vscode的terminal中執(zhí)行
python -m venv env
2.下載 Eigen : 將Eigen解壓到當前目錄命名為 eigen-3.3.8
3.在vscode的terminal中激活虛擬環(huán)境：
./env/Scripts/Activate.ps1
4.安裝pybind11:
pip install pybind11
安裝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" 函數(shù)。程序執(zhí)行將在此處開始并結束。
//
#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中使用的函數(shù)名*/, &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是在同一個文件夾下。
執(zhí)行: "python .\setup.py build_ext --inplace"就會得下面的結果，生成.pyd文件表明我們已經編譯成功。
運行測試：
到此這篇關于python3 通過 pybind11 使用Eigen加速代碼的步驟詳解的文章就介紹到這了,更多相關python3 pybind11 Eigen加速代碼內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了Python 線性回歸分析以及評價指標詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
Python操作PostgreSQL數(shù)據(jù)庫的基本方法(增刪改查)
PostgreSQL數(shù)據(jù)庫是最常用的關系型數(shù)據(jù)庫之一,最吸引人的一點是它作為開源數(shù)據(jù)庫且具有可拓展性,能夠提供豐富的應用,這篇文章主要給大家介紹了關于Python操作PostgreSQL數(shù)據(jù)庫的基本方法,文中介紹了連接PostgreSQL數(shù)據(jù)庫,以及增刪改查,需要的朋友可以參考下2023-09-09
這篇文章主要介紹了Python中字典映射類型的學習教程,是Python入門學習中的基礎知識,需要的朋友可以參考下2015-08-08
這篇文章主要介紹了python paramiko遠程服務器終端操作過程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-12-12
這篇文章主要給大家介紹了關于Python中兩個列表數(shù)字相加的4種方法,我們可以使用Python的加號和減號運算符來實現(xiàn)兩個數(shù)字的相加減,需要的朋友可以參考下2023-08-08
非極大值抑制（Non-Maximum Suppression，NMS），顧名思義就是抑制不是極大值的元素，可以理解為局部最大搜索。這個局部代表的是一個鄰域，鄰域有兩個參數(shù)可變，一是鄰域的維數(shù)，二是鄰域的大小2021-06-06
這篇文章主要介紹了python中sleep函數(shù)用法,實例分析了sleep函數(shù)的功能及使用技巧,非常具有實用價值,需要的朋友可以參考下2015-04-04
flask操作數(shù)據(jù)庫插件Flask-SQLAlchemy的使用
Python?中最廣泛使用的ORM框架是SQLAlchemy,它是一個很強大的關系型數(shù)據(jù)庫框架,本文就來介紹一下flask操作數(shù)據(jù)庫插件Flask-SQLAlchemy的使用,感興趣的可以了解一下2023-09-09
這篇文章主要介紹了Python使用設計模式中的責任鏈模式與迭代器模式的示例,責任鏈模式與迭代器模式都可以被看作為行為型的設計模式,需要的朋友可以參考下2016-03-03
Python 如何創(chuàng)建一個簡單的REST接口
這篇文章主要介紹了Python 如何創(chuàng)建一個簡單的REST接口，文中講解非常細致，代碼幫助大家更好的理解和學習，感興趣的朋友可以了解下2020-07-07