測試Colab提供的免費TPU

從其他網頁上收集到的資訊來看,應該多去使用免費的TPU,因為執行的時間有可能只需要原本的訓練時間的 1/3, 缺點是程式碼需要一點點的微調。參考網頁說明:

Colab提供了免費TPU
https://www.itread01.com/elpqhk.html

參考性佳的範例:
Getting Started with PyTorch on Cloud TPUs
https://medium.com/pytorch/get-started-with-pytorch-cloud-tpus-and-colab-a24757b8f7fc

Get started with our Colab Tutorials

Note: These colab notebooks typically run on small machines (the Compute VMs, which runs the input pipeline) and training is often bottlenecked on the small Compute VM machines. For optimal performance create a GCP VM and TPU pair following our GCP Tutorials:


實際測試結果,在 import xla_model 時,可以看到去初始化 TPU的訊息,最後成功取得 TPU 資源。

TPU 與 cuda 似乎是不同的世界,所以 CUDA 沒被啟用。

相關文章

寫留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *