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}')