測試colab GPU 是否可用

Posted in :

2021-06-04

預設情況，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不可用")