測試colab GPU 是否可用

預設情況,GPU 似乎不會被開啟,請點選「執行階段」裡的「變更執行階段類型」就可以啟用GPU 了。

To enable GPU hardware accelerator, just go to Runtime -> Change runtime type -> Hardware accelerator -> GPU

測試用 script:

import torch
flag = torch.cuda.is_available()
if flag:
    print("CUDA可使用")
else:
    print("CUDA不可用")

相關文章

寫留言

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