Печать доступных графических процессоров и тензоров в Python

Чтобы распечатать все доступные графические процессоры и соответствующие им тензоры в Python, вы можете использовать следующий фрагмент кода:

import torch
if torch.cuda.is_available():
    device_count = torch.cuda.device_count()
    print(f"Number of available GPUs: {device_count}")
    for i in range(device_count):
        device = torch.cuda.get_device_name(i)
        print(f"GPU {i}: {device}")
        tensor = torch.randn(3, 3).cuda(i)  # Example tensor on GPU
        print(f"Tensor on GPU {i}:\n{tensor}")
else:
    print("No GPUs available.")

Этот фрагмент кода проверяет, доступна ли CUDA (вычислительная платформа графического процессора). Если да, он печатает количество доступных графических процессоров и их имена. Он также создает пример тензора на каждом графическом процессоре и печатает значения тензора.