查看 GPU 是否可用，返回 True 则具有能够使用的 GPU
torch.cuda.is_available()
查看本机 GPU 数量
import torch
n_gpu = torch.cuda.device_count()
print(n_gpu)
将模型部署到 GPU
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
0%
