From bf3df097b8763611101e588d053821af3f65d1bf Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 18 Apr 2023 17:14:24 +0530 Subject: [PATCH] Don't use ROCm on Linux if an NVIDIA card is present --- scripts/check_modules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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