Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

ROCm 6.1 switch to stable PyTorch

Browse Source
This commit is contained in:
Disty0
2024-07-25 16:22:14 +03:00
parent 4aabc8b614
commit 918a839952
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
-5
View File
@@ -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")