如何使用TPU
2019/10/06 下午 05:44
機器學習共學討論版
吳乘宇
觀看數:20
回答數:1
收藏數:0
ml100-2
ml100-2-d80
聽說TPU比GPU 快好幾倍,但我用google Colab 選擇TPU
反而比GPU 慢跟CPU差不多。想請教專家,今日作業如何使用TPU來運算?
我用GPU 大概也是算了一個小時左右QQ
ps 因為本人電腦記憶體不夠 kernal都會死掉,所以才轉至google colab執行
回答列表
-
2019/10/09 上午 11:25陳明佑 (Ming You Chen)贊同數:0不贊同數:0留言數:0
同學您好:
我想您直接移植的話, 可能並沒有真的使用到TPU的資源, 所以感覺跟CPU差不多
請參考這一篇內容
https://www.jiqizhixin.com/articles/2018-10-11-5
基本上需要使用特定的函數改寫, 似乎才會使用到TPU資源
因為我們平常也是使用自己機台上的 GPU 執行
因此作業範例並沒有特別針對 colab 改寫, 所以會有這種情形