Add comment.

2026-09-19 09:14:35 +02:00 · 2023-10-08 18:34:48 +02:00
parent 8d35d68167
commit 0bf43ec2cf
1 changed files with 1 additions and 0 deletions
@@ -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}')