PyTorch torch.set_default_device función

Pytorch torch 参考手册Pytorch torch manual de referencia

torch.set_default_deviceEs una función en PyTorch utilizada para establecer el dispositivo predeterminado.

Definición de la función

torch.set_default_device(device)

Ejemplo de uso

Ejemplo

import torch

# Establecer dispositivo predeterminado
if torch.cuda.is_available():
    torch.set_default_device('cuda')
    x = torch.zeros(3, 4)
    print("Dispositivo predeterminado:", x.device)
    torch.set_default_device('cpu')
else:
    print("CUDA no disponible")

Pytorch torch 参考手册Pytorch torch manual de referencia

Otras extensiones