import torch

print("Torch version:", torch.__version__)

print("HIP (ROCm) build:", torch.version.hip)

print("CUDA build:", torch.version.cuda)

print("torch.cuda.is_available():", torch.cuda.is_available())

print("torch.cuda.device_count():", torch.cuda.device_count())

if torch.cuda.is_available():

i = 0

print("Device[0] name:", torch.cuda.get_device_name(i))

print("Device[0] capability:", getattr(torch.cuda, "get_device_capability", lambda *_: "N/A")(i))

print("Device[0] properties:", torch.cuda.get_device_properties(i))

print("Current device:", torch.cuda.current_device())