Fabian
5 lines
263 B
Python
5 lines
263 B
Python
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')
|