原生PyTorch支持,大模型一鍵遷移!寒武紀(jì)開源Torch-MLU
AIGC動(dòng)態(tài)歡迎閱讀
原標(biāo)題:原生PyTorch支持,大模型一鍵遷移!寒武紀(jì)開源Torch-MLU
關(guān)鍵字:開發(fā)者,設(shè)備,社區(qū),插件,版本
文章來(lái)源:智東西
內(nèi)容字?jǐn)?shù):0字
內(nèi)容摘要:
寒武紀(jì)開源了Torch-MLU,并實(shí)現(xiàn)了寒武紀(jì)硬件對(duì)于PyTorch的原生支持。
今日,寒武紀(jì)開源了PyTorch設(shè)備后端擴(kuò)展插件Torch-MLU,并實(shí)現(xiàn)了寒武紀(jì)硬件對(duì)于PyTorch的原生支持,充分提升了開發(fā)者的使用體驗(yàn)和集成效率。
早在2018年,寒武紀(jì)就開始了與PyTorch框架的集成適配工作。自PyTorch 1.3版本開始就支持寒武紀(jì)MLU系列智能加速卡作為PyTorch的加速后端,使開發(fā)者能夠?qū)⒃然贕PU的深度學(xué)習(xí)網(wǎng)絡(luò)高效遷移到基于寒武紀(jì)MLU的PyTorch環(huán)境中。
隨著PyTorch社區(qū)發(fā)布PyTorch 2.4版本,配合新版本(≥ V1.22)的Torch-MLU插件,寒武紀(jì)硬件實(shí)現(xiàn)了對(duì)PyTorch的原生支持。開發(fā)者無(wú)需再安裝寒武紀(jì)定制的Cambricon PyTorch,而是可以基于社區(qū)的PyTorch安裝Torch-MLU插件,就可將寒武紀(jì)硬件的算力接入原生PyTorch。
與此同時(shí),Torch-MLU的開源使寒武紀(jì)能夠更快地響應(yīng)開發(fā)者的反饋和問(wèn)題,搭建了高效的溝通橋梁,促進(jìn)了與全球開發(fā)者的協(xié)作和知識(shí)共享。寒武紀(jì)通過(guò)持續(xù)貢獻(xiàn)開源社區(qū)的實(shí)際行動(dòng),展示著擁抱開
原文鏈接:原生PyTorch支持,大模型一鍵遷移!寒武紀(jì)開源Torch-MLU
聯(lián)系作者
文章來(lái)源:智東西
作者微信:
作者簡(jiǎn)介: