溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

怎么在python中利用Eigen對代碼進行加速

發(fā)布時間：2020-12-07 14:21:07 來源：億速云 閱讀：504 作者：Leah 欄目：開發(fā)技術(shù)

這期內(nèi)容當中小編將會給大家?guī)碛嘘P怎么在python中利用Eigen對代碼進行加速，文章內(nèi)容豐富且以專業(yè)的角度為大家分析和敘述，閱讀完這篇文章希望大家可以有所收獲。

第一步：準備系統(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解壓到當前目錄命名為 eigen-3.3.8
3.在vscode的terminal中激活虛擬環(huán)境：

 ./env/Scripts/Activate.ps1

怎么在python中利用Eigen對代碼進行加速

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);
}

第五步：設置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])

第六步：編譯測試

怎么在python中利用Eigen對代碼進行加速

注意：我的cpp_python.cpp和setup.py是在同一個文件夾下。

執(zhí)行: "python .\setup.py build_ext --inplace"就會得下面的結(jié)果，生成.pyd文件表明我們已經(jīng)編譯成功。

怎么在python中利用Eigen對代碼進行加速

運行測試：

怎么在python中利用Eigen對代碼進行加速

上述就是小編為大家分享的怎么在python中利用Eigen對代碼進行加速了，如果剛好有類似的疑惑，不妨參照上述分析進行理解。如果想知道更多相關知識，歡迎關注億速云行業(yè)資訊頻道。

向AI問一下細節(jié)

免責聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手