logo
Loading...

如何使用TPU - Cupoy

聽說TPU比GPU 快好幾倍,但我用google Colab 選擇TPU反而比GPU 慢跟CPU差不...

ml100-2,ml100-2-d80

如何使用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 改寫, 所以會有這種情形