From 0bf43ec2cf567c684ee089246816af21652fc918 Mon Sep 17 00:00:00 2001 From: Hameer Abbasi Date: Sun, 8 Oct 2023 18:34:48 +0200 Subject: [PATCH] Add comment. --- installer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/installer.py b/installer.py index 9f255192e..f700b63e2 100644 --- a/installer.py +++ b/installer.py @@ -404,6 +404,7 @@ def check_torch(): except Exception as e: log.debug(f'ROCm hipconfig failed: {e}') rocm_ver = None + # ROCm 5.7 and above don't have PyTorch 2.1.0. if rocm_ver_tuple[:2] <= (5, 6): # install torch nightly via torchvision to avoid wasting bandwidth when torchvision depends on torch from yesterday torch_command = os.environ.get('TORCH_COMMAND', f'torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm{rocm_ver}')