python 通過(guò) pybind11 使用Eigen加速代碼的步驟
python是很容易上手的編程語(yǔ)言，但是有些時(shí)候使用python編寫(xiě)的程序并不能保證其運(yùn)行速度（例如：while 和 for），這個(gè)時(shí)候我們就需要借助c++等為我們的代碼提速。下面是我使用pybind11調(diào)用c++的Eigen庫(kù)的簡(jiǎn)單介紹:
第一步：準(zhǔn)備系統(tǒng)和IDE：
Windows 10
vs2015 (用于調(diào)試c++代碼)
vscode (調(diào)試python代碼)
第二步：python虛擬環(huán)境：
1.創(chuàng)建虛擬python虛擬環(huán)境： 在vscode的terminal中執(zhí)行
python -m venv env
2.下載 Eigen: 將Eigen解壓到當(dāng)前目錄命名為 eigen-3.3.8
3.在vscode的terminal中激活虛擬環(huán)境：
./env/Scripts/Activate.ps1
4.安裝pybind11:
pip install pybind11
5.安裝numpy==1.19.3（使用1.19.4可能會(huì)有問(wèn)題）
pip install numpy==1.19.3
第三步：使用vs2015編寫(xiě)cpp_python.cpp, 并保證沒(méi)有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í)行將在此處開(kāi)始并結(jié)束。
//
#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";//解釋說(shuō)明
m.def("mat_add_py"/*在pyhon中使用的函數(shù)名*/, &add_mat);
}
第五步：設(shè)置setup.py用來(lái)編譯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', # 依賴的第三方庫(kù)的頭文件
r'.\env\Lib\site-packages\pybind11\include']
)
setup(ext_modules=[add_mat_module])
第六步：編譯測(cè)試
注意：我的cpp_python.cpp和setup.py是在同一個(gè)文件夾下。
執(zhí)行: "python .\setup.py build_ext --inplace"就會(huì)得下面的結(jié)果，生成.pyd文件表明我們已經(jīng)編譯成功。
運(yùn)行測(cè)試：
以上就是python 通過(guò) pybind11 使用Eigen加速代碼的步驟的詳細(xì)內(nèi)容，更多關(guān)于python 加速代碼的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
使用Python實(shí)現(xiàn)with結(jié)構(gòu)的@contextmanager方法詳解
這篇文章主要介紹了使用Python實(shí)現(xiàn)with結(jié)構(gòu)的@contextmanager方法詳解,這個(gè)結(jié)構(gòu)的好處，一個(gè)是簡(jiǎn)潔，一個(gè)是當(dāng)我們對(duì)文件操作的邏輯很長(zhǎng)的時(shí)候，不會(huì)因?yàn)橥岁P(guān)閉文件而造成不必要的錯(cuò)誤,需要的朋友可以參考下2023-07-07
python實(shí)現(xiàn)通過(guò)shelve修改對(duì)象實(shí)例
這篇文章主要介紹了python實(shí)現(xiàn)通過(guò)shelve修改對(duì)象實(shí)例,是一個(gè)比較實(shí)用的技巧,需要的朋友可以參考下2014-09-09
Python與數(shù)據(jù)庫(kù)交互：入門(mén)指南
這篇文章主要介紹了Python與數(shù)據(jù)庫(kù)交互：入門(mén)指南的相關(guān)資料,需要的朋友可以參考下2023-11-11
關(guān)于SSD目標(biāo)檢測(cè)模型的人臉口罩識(shí)別
這篇文章主要介紹了關(guān)于SSD目標(biāo)檢測(cè)模型的人臉口罩識(shí)別問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-11-11
Python Tkinter 簡(jiǎn)單登錄界面的實(shí)現(xiàn)
今天小編就為大家分享一篇Python Tkinter 簡(jiǎn)單登錄界面的實(shí)現(xiàn)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06
Python facenet進(jìn)行人臉識(shí)別測(cè)試過(guò)程解析
這篇文章主要介紹了Python facenet進(jìn)行人臉識(shí)別測(cè)試過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08
OpenCV半小時(shí)掌握基本操作之角點(diǎn)檢測(cè)
這篇文章主要介紹了OpenCV基本操作之角點(diǎn)檢測(cè)，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-09-09
python文字轉(zhuǎn)語(yǔ)音的實(shí)例代碼分析
在本篇文章里小編給大家整理的是關(guān)于python文字轉(zhuǎn)語(yǔ)音的實(shí)例代碼分析，有需要的朋友們可以參考下。2019-11-11
使用python+requests+pytest實(shí)現(xiàn)接口自動(dòng)化
這篇文章主要介紹了使用python+requests+pytest實(shí)現(xiàn)接口自動(dòng)化,在當(dāng)前互聯(lián)網(wǎng)產(chǎn)品迭代頻繁的背景下，回歸測(cè)試的時(shí)間越來(lái)越少，但接口自動(dòng)化測(cè)試因其實(shí)現(xiàn)簡(jiǎn)單、維護(hù)成本低，容易提高覆蓋率等特點(diǎn)，越來(lái)越受重視,需要的朋友可以參考下2023-08-08
iloc[]函數(shù)屬于pandas庫(kù)全稱為index?location，即對(duì)數(shù)據(jù)進(jìn)行位置索引，從而在數(shù)據(jù)表中提取出相應(yīng)的數(shù)據(jù)，本文通過(guò)實(shí)例代碼介紹pandas中iloc()函數(shù)，感興趣的朋友一起看看吧2023-04-04