PyTorch torch.get_default_device 函数

Pytorch torch 参考手册 Pytorch torch 参考手册

torch.get_default_device 是 PyTorch 中用于获取当前默认设备的函数。

函数定义

torch.get_default_device()

使用示例

实例

import torch

device = torch.get_default_device()
print("默认设备:", device)

Pytorch torch 参考手册 Pytorch torch 参考手册