diff --git a/scripts/check_modules.py b/scripts/check_modules.py index 2b47527f..0abf754e 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -82,7 +82,7 @@ def apply_torch_install_overrides(module_version: str): elif os_name == "Linux": with open("/proc/bus/pci/devices", "r") as f: device_info = f.read() - if "amdgpu" in device_info: + if "amdgpu" in device_info and "nvidia" not in device_info: index_url = "https://download.pytorch.org/whl/rocm5.4.2" return module_version, index_url