最近在学Pytorch,在这里做一下记录
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# 确认我们的电脑支持CUDA,然后显示CUDA信息:
print(device)
参考资料:
pytorch-handbook
最近在学Pytorch,在这里做一下记录
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# 确认我们的电脑支持CUDA,然后显示CUDA信息:
print(device)
参考资料:
pytorch-handbook