2023年10月5日(北京時間),PyTorch社區(qū)正式發(fā)布2.1版本,經(jīng)過PyTorch社區(qū)與昇騰的持續(xù)合作和共同努力,Pytorch 2.1版本已同步支持昇騰NPU,意味著開發(fā)者可直接在PyTorch 2.1上基于昇騰進(jìn)行模型開發(fā)。未來,通過持續(xù)的社區(qū)技術(shù)貢獻(xiàn),昇騰將與PyTorch社區(qū)版本同步演進(jìn)及發(fā)布,共促AI技術(shù)創(chuàng)新和開源生態(tài)發(fā)展。
圖注:PyTorch 社區(qū)2.1版本的官方發(fā)布聲明中宣布昇騰NPU已成功對接
PyTorch在新版本中發(fā)布了更加完善的第三方設(shè)備接入機(jī)制,該特性由昇騰主導(dǎo),與PyTorch社區(qū)的核心維護(hù)者共同完成?；谠撎匦?三方AI算力設(shè)備無需對原有框架代碼進(jìn)行修改就能對接PyTorch框架,昇騰也提供了官方認(rèn)證的Torch NPU參考實現(xiàn),可以指導(dǎo)三方設(shè)備便捷接入。
在早前版本中,由于缺少對于昇騰算力設(shè)備的原生支持,開發(fā)者如果想在主流框架下調(diào)用昇騰NPU實現(xiàn)加速,還需要花費數(shù)周時間進(jìn)行模型遷移。而現(xiàn)在,基于新版本,用戶可以在昇騰NPU上直接享受原生PyTorch的開發(fā)體驗,獲得可以高效運行在昇騰算力設(shè)備上的模型和應(yīng)用。
基于PyTorch,昇騰已經(jīng)適配了BLOOM、GPT-3、LLaMA等業(yè)界主流大模型,深度優(yōu)化后性能可持平業(yè)界,在最新的PyTorch版本上,昇騰將與社區(qū)緊密合作,同步演進(jìn),不斷提升模型加速能力。
未來,昇騰將通過持續(xù)的技術(shù)貢獻(xiàn),積極推動PyTorch社區(qū)多樣性算力改進(jìn),優(yōu)化PyTorch框架,幫助釋放昇騰算力,與PyTorch社區(qū)貢獻(xiàn)者一起,共促AI技術(shù)創(chuàng)新和開源生態(tài)的快速發(fā)展。
