Back to top
  • 공유 分享
  • 인쇄 列印
  • 글자크기 字體大小
已複製網址

Google將TPU列為PyTorch原生支援硬體,與GPU、CPU並列

筆電螢幕上顯示的開發程式碼視窗 / TokenPost.ai

Google($GOOGL)公開與Meta($META)的PyTorch團隊共同開發的開源專案TorchTPU,讓TPU成為PyTorch原生支援的硬體選項,與GPU、CPU並列。

Google Cloud官方部落格表示,Google在「PyTorch Conference 2026」上將TorchTPU開源,程式碼已公開於GitHub。現有PyTorch程式碼只需少量修改,就能移植至TPU。

Google Cloud也指出,vLLM、SGLang、TorchTitan、VeRL及Ray等框架已與TorchTPU整合。在Hugging Face Transformers v5中,設定device_map="tpu"即可將模型部署至TPU。

<版權所有 ⓒ TokenPost,未經授權禁止轉載與散佈>

最受歡迎

其他相關文章

留言 0

留言小技巧

好文章。 希望有後續報導。 分析得很棒。

0/1000

留言小技巧

好文章。 希望有後續報導。 分析得很棒。
1