import torch
print("Device props:", torch.cuda.get_device_properties(0))
print("Arch list:", torch.cuda.get_arch_list())
print("GCN arch:", torch.cuda.get_device_properties(0).gcnArchName if hasattr(torch.cuda.get_device_properties(0), 'gcnArchName') else 'N/A')
