這篇文章將為大家詳細(xì)講解有關(guān)numba編譯器是什么,文章內(nèi)容質(zhì)量較高,因此小編分享給大家做個(gè)參考,希望大家閱讀完這篇文章后對(duì)相關(guān)知識(shí)有一定的了解。
創(chuàng)新互聯(lián)公司從2013年成立,是專業(yè)互聯(lián)網(wǎng)技術(shù)服務(wù)公司,擁有項(xiàng)目成都網(wǎng)站制作、網(wǎng)站建設(shè)網(wǎng)站策劃,項(xiàng)目實(shí)施與項(xiàng)目整合能力。我們以讓每一個(gè)夢(mèng)想脫穎而出為使命,1280元峰峰礦做網(wǎng)站,已為上家服務(wù),為峰峰礦各地企業(yè)和個(gè)人服務(wù),聯(lián)系電話:18982081108
什么是numba?
numba是一款可以將python函數(shù)編譯為機(jī)器代碼的JIT編譯器,經(jīng)過(guò)numba編譯的python代碼(僅限數(shù)組運(yùn)算),其運(yùn)行速度可以接近C或FORTRAN語(yǔ)言。
numba工作流程
python之所以慢,是因?yàn)樗强緾Python編譯的,numba的作用是給python換一種編譯器。
python、c、numba三種編譯器速度對(duì)比
使用numba非常簡(jiǎn)單,只需要將numba裝飾器應(yīng)用到python函數(shù)中,無(wú)需改動(dòng)原本的python代碼,numba會(huì)自動(dòng)完成剩余的工作。
import numba from numba import jit @jit(nopython=True) # jit,numba裝飾器中的一種 def go_fast(a): # 首次調(diào)用時(shí),函數(shù)被編譯為機(jī)器代碼 trace = 0 # 假設(shè)輸入變量是numpy數(shù)組 for i in range(a.shape[0]): # Numba 擅長(zhǎng)處理循環(huán) trace += np.tanh(a[i, i]) return a + trace
以上代碼是一個(gè)python函數(shù),用以計(jì)算numpy數(shù)組各個(gè)數(shù)值的雙曲正切值,我們使用了numba裝飾器,它將這個(gè)python函數(shù)編譯為等效的機(jī)器代碼,可以大大減少運(yùn)行時(shí)間。
numba適合科學(xué)計(jì)算
numpy是為面向numpy數(shù)組的計(jì)算任務(wù)而設(shè)計(jì)的。
在面向數(shù)組的計(jì)算任務(wù)中,數(shù)據(jù)并行性對(duì)于像GPU這樣的加速器是很自然的。Numba了解NumPy數(shù)組類型,并使用它們生成高效的編譯代碼,用于在GPU或多核CPU上執(zhí)行。特殊裝飾器還可以創(chuàng)建函數(shù),像numpy函數(shù)那樣在numpy數(shù)組上廣播。
什么情況下使用numba呢?
·使用numpy數(shù)組做大量科學(xué)計(jì)算時(shí)
·使用for循環(huán)時(shí)
學(xué)習(xí)使用numba
第一步:導(dǎo)入numpy、numba及其編譯器
import numpy as np import numba from numba import jit
第二步:傳入numba裝飾器jit,編寫(xiě)函數(shù)
# 傳入jit,numba裝飾器中的一種 @jit(nopython=True) def go_fast(a): # 首次調(diào)用時(shí),函數(shù)被編譯為機(jī)器代碼 trace = 0 # 假設(shè)輸入變量是numpy數(shù)組 for i in range(a.shape[0]): # Numba 擅長(zhǎng)處理循環(huán) trace += np.tanh(a[i, i]) # numba喜歡numpy函數(shù) return a + trace # numba喜歡numpy廣播
nopython = True選項(xiàng)要求完全編譯該函數(shù)(以便完全刪除Python解釋器調(diào)用),否則會(huì)引發(fā)異常。這些異常通常表示函數(shù)中需要修改的位置,以實(shí)現(xiàn)優(yōu)于Python的性能。強(qiáng)烈建議您始終使用nopython = True。
第三步:給函數(shù)傳遞實(shí)參
# 因?yàn)楹瘮?shù)要求傳入的參數(shù)是nunpy數(shù)組 x = np.arange(100).reshape(10, 10) # 執(zhí)行函數(shù) go_fast(x)
第四步:經(jīng)numba加速的函數(shù)執(zhí)行時(shí)間
% timeit go_fast(x)
輸出:
3.63 μs ± 156 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
第五步:不經(jīng)numba加速的函數(shù)執(zhí)行時(shí)間
def go_fast(a): # 首次調(diào)用時(shí),函數(shù)被編譯為機(jī)器代碼 trace = 0 # 假設(shè)輸入變量是numpy數(shù)組 for i in range(a.shape[0]): # Numba 擅長(zhǎng)處理循環(huán) trace += np.tanh(a[i, i]) # numba喜歡numpy函數(shù) return a + trace # numba喜歡numpy廣播 x = np.arange(100).reshape(10, 10) %timeit go_fast(x)
輸出:
136 μs ± 1.09 μs per loop (mean ± std. dev. of 7 runs, 10000 loops each)
結(jié)論:
在numba加速下,代碼執(zhí)行時(shí)間為3.63微秒/循環(huán)。不經(jīng)過(guò)numba加速,代碼執(zhí)行時(shí)間為136微秒/循環(huán),兩者相比,前者快了40倍。
numba讓python飛起來(lái)
前面已經(jīng)對(duì)比了numba使用前后,python代碼速度提升了40倍,但這還不是最快的。
這次,我們不使用numpy數(shù)組,僅用for循環(huán),看看nunba對(duì)for循環(huán)到底有多鐘愛(ài)!
# 不使用numba的情況 def t(): x = 0 for i in np.arange(5000): x += i return x %timeit(t())
輸出:
408 μs ± 9.73 μs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
# 使用numba的情況 @jit(nopython=True) def t(): x = 0 for i in np.arange(5000): x += i return x %timeit(t())
輸出:
1.57 μs ± 53.8 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
結(jié)論:使用numba前后分別是408微秒/循環(huán)、1.57微秒/循環(huán),速度整整提升了200多倍!
結(jié)語(yǔ)
numba對(duì)python代碼運(yùn)行速度有巨大的提升,這極大的促進(jìn)了大數(shù)據(jù)時(shí)代的python數(shù)據(jù)分析能力,對(duì)數(shù)據(jù)科學(xué)工作者來(lái)說(shuō),這真是一個(gè)lucky tool !
當(dāng)然numba不會(huì)對(duì)numpy和for循環(huán)以外的python代碼有很大幫助,你不要指望numba可以幫你加快從數(shù)據(jù)庫(kù)取數(shù),這點(diǎn)它真的做不到哈。
關(guān)于numba編譯器是什么就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,可以學(xué)到更多知識(shí)。如果覺(jué)得文章不錯(cuò),可以把它分享出去讓更多的人看到。
網(wǎng)站標(biāo)題:numba編譯器是什么
轉(zhuǎn)載源于:http://www.rwnh.cn/article36/jiecpg.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供網(wǎng)站建設(shè)、網(wǎng)站導(dǎo)航、外貿(mào)建站、用戶體驗(yàn)、標(biāo)簽優(yōu)化、建站公司
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請(qǐng)盡快告知,我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如需處理請(qǐng)聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來(lái)源: 創(chuàng)新互聯(lián)