wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 13:02:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Restrict Windows ROCm SDK env setup to ROCm builds

Browse Source
This commit is contained in:
野生の男 2026-04-03 01:26:15 +09:00
parent b6a99158b5
commit eb465ad574
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
main.py
View File

@ -31,10 +31,21 @@ if __name__ == "__main__":
setup_logger(log_level=args.verbose, use_stdout=args.log_stdout) setup_logger(log_level=args.verbose, use_stdout=args.log_stdout)
def is_rocm_torch_build():
try:
torch_version = importlib.metadata.version("torch").lower()
except importlib.metadata.PackageNotFoundError:
return False
return "rocm" in torch_version
def configure_windows_rocm_sdk_env(): def configure_windows_rocm_sdk_env():
if os.name != "nt": if os.name != "nt":
return None return None
if not is_rocm_torch_build():
return None
if any(os.environ.get(var) for var in ("ROCM_HOME", "ROCM_PATH", "HIP_PATH")): if any(os.environ.get(var) for var in ("ROCM_HOME", "ROCM_PATH", "HIP_PATH")):
return None return None