From 1cc373fd89d68b75d7d4c9cc2e7dc6d8f6603c86 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 17 May 2024 23:03:52 +0300 Subject: [PATCH] Simplify minimum ROCm version check --- installer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/installer.py b/installer.py index d726d3eb4..fc93016b8 100644 --- a/installer.py +++ b/installer.py @@ -532,7 +532,7 @@ def check_torch(): torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.0') elif rocm_ver == "6.1": # need nightlies torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm6.1') - elif int(rocm_ver.rsplit(".")[0]) <= 4 or (int(rocm_ver.rsplit(".")[0]) == 5 and int(rocm_ver.rsplit(".")[1]) <= 5): # oldest supported version is 5.5 + elif float(rocm_ver) < 5.5: # oldest supported version is 5.5 log.warning(f"Unsupported ROCm version detected: {rocm_ver}") log.warning("Minimum supported ROCm version is 5.5") torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm5.5')