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