Fonction torch.set_default_device de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.set_default_deviceIl s'agit d'une fonction de PyTorch utilisée pour définir l'appareil par défaut.

Définition de fonction

torch.set_default_device(device)

Exemple d'utilisation

Exemple

import torch

# Définir l'appareil par défaut
if torch.cuda.is_available():
    torch.set_default_device('cuda')
    x = torch.zeros(3, 4)
    print("Default device:", x.device)
    torch.set_default_device('cpu')
else:
    print("CUDA n'est pas disponible")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions