limit architecture
@@ -188,6 +188,9 @@ library_records['nccl'] = _nccl_records
def install_lib(cuda, prefix, library):
+ if os.uname().machine != 'x86_64':
+ raise RuntimeError('''
+Currently this tool only supports x86_64 architecture.''')
record = None
lib_records = library_records
for record in lib_records[library]:
