From 900264a194f440fce4e1720888e17e3c78fbca1d Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Wed, 5 Jun 2024 10:38:03 +0900 Subject: [PATCH] windows conceal rocm if zluda disabled --- installer.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/installer.py b/installer.py index 88f50e5d4..9f23e4786 100644 --- a/installer.py +++ b/installer.py @@ -540,6 +540,16 @@ def check_torch(): log.info("For ZLUDA support specify '--use-zluda'") log.info('Using CPU-only torch') torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision') + + # conceal ROCm installed + os.environ.pop("ROCM_HOME", None) + os.environ.pop("ROCM_PATH", None) + paths = os.environ["PATH"].split(";") + paths_no_rocm = [] + for path in paths: + if "ROCm" not in path: + paths_no_rocm.append(path) + os.environ["PATH"] = ";".join(paths_no_rocm) else: if rocm_ver is None: # assume the latest if version check fails torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.0')