From 918a839952bec0b576c078c39144b02c6ff97ec0 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 25 Jul 2024 16:22:14 +0300 Subject: [PATCH] ROCm 6.1 switch to stable PyTorch --- installer.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/installer.py b/installer.py index cb6ee674a..38e8ec6de 100644 --- a/installer.py +++ b/installer.py @@ -532,11 +532,6 @@ def install_rocm_zluda(torch_command): else: if rocm.version is None: # assume the latest if version check fails torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.0') - elif rocm.version == "6.1": # need nightlies - if args.experimental: - torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm6.1') - else: - torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.0') elif float(rocm.version) < 5.5: # oldest supported version is 5.5 log.warning(f"Unsupported ROCm version detected: {rocm.version}") log.warning("Minimum supported ROCm version is 5.5")