Fonction torch.device de PyTorch

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

torch.deviceest une fonction de PyTorch utilisée pour créer des objets de type device, utilisée pour spécifier le périphérique sur lequel les tenseurs sont stockés.

Définition de fonction

torch.device(device)

Exemples d'utilisation

Exemples

import torch

# Créer un périphérique CPU
cpu_device = torch.device('cpu')
print("Périphérique CPU :", cpu_device)

# Créer un périphérique CUDA
if torch.cuda.is_available():
    cuda_device = torch.device('cuda:0')
    print("Périphérique CUDA :", cuda_device)

    # Créer un tenseur sur GPU
    x = torch.zeros(3, 4, device=cuda_device)
    print("Périphérique du tenseur :", x.device)

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

Autres extensions