Uploaded sanitized BC250/ROCm Repository.
This commit is contained in:
@@ -0,0 +1,4 @@
|
||||
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')
|
||||
Reference in New Issue
Block a user