PyTorch torch.get_default_device 函数
Pytorch torch 参考手册torch.get_default_device 是 PyTorch 中用于获取当前默认设备的函数。
函数定义
torch.get_default_device()
使用示例
实例
import torch
device = torch.get_default_device()
print("默认设备:", device)
device = torch.get_default_device()
print("默认设备:", device)