【PyTorch】查看自己的電腦是否已經準備好GPU加速（CUDA）

我們可以使用如下幾行簡單的代碼來進行確認：

import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)

那麼在CUDA已經準備好的電腦上，會輸出：

cuda:0

而在沒有CUDA的電腦上則輸出：

cpu