PyTorch+昇騰 共促AI生態(tài)創(chuàng)新發(fā)展

2023年10月5日(北京時(shí)間),PyTorch社區(qū)正式發(fā)布2.1版本,經(jīng)過(guò)PyTorch社區(qū)與昇騰的持續(xù)合作和共同努力,Pytorch 2.1版本已同步支持昇騰NPU,意味著開(kāi)發(fā)者可直接在PyTorch 2.1上基于昇騰進(jìn)行模型開(kāi)發(fā)。未來(lái),通過(guò)持續(xù)的社區(qū)技術(shù)貢獻(xiàn),昇騰將與PyTorch社區(qū)版本同步演進(jìn)及發(fā)布,共促AI技術(shù)創(chuàng)新和開(kāi)源生態(tài)發(fā)展。

圖注:PyTorch 社區(qū)2.1版本的官方發(fā)布聲明中宣布昇騰NPU已成功對(duì)接

PyTorch在新版本中發(fā)布了更加完善的第三方設(shè)備接入機(jī)制,該特性由昇騰主導(dǎo),與PyTorch社區(qū)的核心維護(hù)者共同完成?;谠撎匦?,三方AI算力設(shè)備無(wú)需對(duì)原有框架代碼進(jìn)行修改就能對(duì)接PyTorch框架,昇騰也提供了官方認(rèn)證的Torch NPU參考實(shí)現(xiàn),可以指導(dǎo)三方設(shè)備便捷接入。

在早前版本中,由于缺少對(duì)于昇騰算力設(shè)備的原生支持,開(kāi)發(fā)者如果想在主流框架下調(diào)用昇騰NPU實(shí)現(xiàn)加速,還需要花費(fèi)數(shù)周時(shí)間進(jìn)行模型遷移。而現(xiàn)在,基于新版本,用戶可以在昇騰NPU上直接享受原生PyTorch的開(kāi)發(fā)體驗(yàn),獲得可以高效運(yùn)行在昇騰算力設(shè)備上的模型和應(yīng)用。

基于PyTorch,昇騰已經(jīng)適配了BLOOM、GPT-3、LLaMA等業(yè)界主流大模型,深度優(yōu)化后性能可持平業(yè)界,在最新的PyTorch版本上,昇騰將與社區(qū)緊密合作,同步演進(jìn),不斷提升模型加速能力。

未來(lái),昇騰將通過(guò)持續(xù)的技術(shù)貢獻(xiàn),積極推動(dòng)PyTorch社區(qū)多樣性算力改進(jìn),優(yōu)化PyTorch框架,幫助釋放昇騰算力,與PyTorch社區(qū)貢獻(xiàn)者一起,共促AI技術(shù)創(chuàng)新和開(kāi)源生態(tài)的快速發(fā)展。

(免責(zé)聲明:本網(wǎng)站內(nèi)容主要來(lái)自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準(zhǔn)確性及可靠性,但不保證有關(guān)資料的準(zhǔn)確性及可靠性,讀者在使用前請(qǐng)進(jìn)一步核實(shí),并對(duì)任何自主決定的行為負(fù)責(zé)。本網(wǎng)站對(duì)有關(guān)資料所引致的錯(cuò)誤、不確或遺漏,概不負(fù)任何法律責(zé)任。
任何單位或個(gè)人認(rèn)為本網(wǎng)站中的網(wǎng)頁(yè)或鏈接內(nèi)容可能涉嫌侵犯其知識(shí)產(chǎn)權(quán)或存在不實(shí)內(nèi)容時(shí),應(yīng)及時(shí)向本網(wǎng)站提出書(shū)面權(quán)利通知或不實(shí)情況說(shuō)明,并提供身份證明、權(quán)屬證明及詳細(xì)侵權(quán)或不實(shí)情況證明。本網(wǎng)站在收到上述法律文件后,將會(huì)依法盡快聯(lián)系相關(guān)文章源頭核實(shí),溝通刪除相關(guān)內(nèi)容或斷開(kāi)相關(guān)鏈接。 )