首页
/
每日頭條
/
生活
/
谷歌AI加速器CloudTPU進一步支持PyTorch框架
谷歌AI加速器CloudTPU進一步支持PyTorch框架
更新时间:2024-11-16 20:58:03

近日,谷歌宣布雲端加速器TPU開始進一步支持深度學習函式庫PyTorch,僅需使用Python套件PyTorch/XLA,進行最小程度的修改,就能夠使用TPU來加速機器學習工作負載,且在TPU還不支援的操作,自動退回到以CPU運算,同時PyTorch/XLA也會生成詳細的報告,幫助使用者發現應用程序瓶頸,高效執行機器學習工作負載。

XLA是Google在2018年推出的最佳化編譯器,通過最佳化算法運算,可以提升機器學習模型的執行速度,XLA支援處理器以及繪圖卡,也能夠在Google TPU上執行,并可以讓Cloud TPU更容易支持PyTorch。

谷歌AI加速器CloudTPU進一步支持PyTorch框架(谷歌AI加速器CloudTPU進一步支持PyTorch框架)1

在最新的PyTorch/XLA上,使Cloud TPU正式支援PyTorch 1.6,在基礎設施層提供模型平行運算能力,允許開發者可以在多個TPU核心上,分散大規模嵌入表,因此有能力創建許多過去無法實現的應用。

另外,谷歌還在雲端上發布的深度學習虛拟機器(DLVM)映像檔,其中包括PyTorch等各種深度學習框架,而 PyTorch/XLA 1.6現在預裝在DLVM中,并且已經對Cloud TPU進行最佳化。

為了讓開發者更好上手PyTorch/XLA,Google雲端還提供一組常用深度學習模型開源實作,以及相關的教程,這些模型包括使用ImageNet資料集的圖像分類任務ResNet-50,還有Transformer、RoBERTa以及BERT等知名模型,也能使用PyTorch 1.6才加入,由臉書開發的深度學習推薦模型(DLRM)。官方提到,在Cloud TPU訓練這些模型,大部分的情況僅需修改少量代碼即可實現。

,
Comments
Welcome to tft每日頭條 comments! Please keep conversations courteous and on-topic. To fosterproductive and respectful conversations, you may see comments from our Community Managers.
Sign up to post
Sort by
Show More Comments
推荐阅读
葉非墨溫暖是哪部小說
葉非墨溫暖是哪部小說
葉非墨溫暖是哪部小說?葉非墨溫暖是《總裁的替身前妻》這部小說的主人公,我來為大家講解一下關于葉非墨溫暖是哪部小說?跟着小編一起來看一看吧!葉非墨溫暖是哪部小說葉非墨溫暖是《總裁的替身前妻》這部小說的主人公。《總裁的替身前妻》簡介:葉非墨溫暖...
2024-11-16
新手入門滑闆合适的幾個動作
新手入門滑闆合适的幾個動作
滑闆,相信已經不少人都多多少少有了解和接觸過,也有許多人準備打算入手一塊屬于自己的滑闆,想要體驗一下滑闆這項運動的魅力,那麼,本文将為你提供一些幫助和建議,避免一些不必要的尴尬和受傷。關于滑闆的分類滑闆的有三種類型,各自的闆型都不盡相同,分...
2024-11-16
痛經喝紅糖水有效麼
痛經喝紅糖水有效麼
痛經是非常常見的現象,每個月都會來那麼幾天,讓女性朋友們感覺非常的痛苦,不少女性在出現痛經症狀的時候,隻會和紅糖水,及時緩解痛經的方法很多的,今天就給大家介紹幾種,痛經還在喝紅糖水?這幾種方法也不錯!引起痛經的原因主要有:(1)子宮頸管狹窄...
2024-11-16
縫針後疤痕怎麼修複有效
縫針後疤痕怎麼修複有效
導緻疤痕出現的原因有很多,比較常見的一種就是因為縫針留下的,這些縫針疤痕也是能解決的,但這還涉及到縫針疤痕祛除的最佳時間,如果能在合适的機會去除掉疤痕的話,效果肯定是最佳的,所以把握時間是最重要的,那麼,縫針疤痕祛除最佳時間是什麼時候?1....
2024-11-16
那些年我們青春剛剛好
那些年我們青春剛剛好
《那些年,我們正年輕》聚焦年輕人的奮鬥與情感。廣州日報訊(全媒體記者曾俊)聚焦“兩彈一星”科研建設的年代大戲《那些年,我們正年輕》正在北京衛視、騰訊視頻熱播,收獲了一批年輕觀衆。該劇講述了上世紀60年代至90年代中國一代航天人的青春奮鬥故事...
2024-11-16
Copyright 2023-2024 - www.tftnews.com All Rights Reserved