From 7eb7dd587dc5df9fa5d5cd2ad9e23b1337c3d809 Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Thu, 18 Jul 2024 01:00:50 +0900 Subject: [PATCH] fix --- installer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/installer.py b/installer.py index 80acacab9..6c62b897d 100644 --- a/installer.py +++ b/installer.py @@ -565,7 +565,7 @@ def install_rocm_zluda(torch_command): continue supported_archs.append(file[6:-3]) for gpu in hip_visible_devices: - if gpu not in supported_archs: + if gpu[1] not in supported_archs: hipblaslt_available = False break log.debug(f'hipBLASLt supported_archs={supported_archs}, available={hipblaslt_available}')