Чтобы распечатать все доступные графические процессоры и соответствующие им тензоры в 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 (вычислительная платформа графического процессора). Если да, он печатает количество доступных графических процессоров и их имена. Он также создает пример тензора на каждом графическом процессоре и печатает значения тензора.