Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-17 16:24:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

ROCm 6.4 support with --use-nightly

Browse Source
This commit is contained in:
Disty0
2025-06-10 13:42:43 +03:00
parent d2ffee1b4e
commit a6b58efe45
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CHANGELOG.md
+2 -1
View File
@@ -12,7 +12,8 @@
- don't override user set gfx version with ROCm
- **Torch**
- set default to `torch==2.7.1`
- Set default to `torch==2.7.1`
- Support ROCm 6.4 with `---use-nightly`
- **SDNQ Quantization**
- Add group size support for convolutional layers
installer.py
+3 -1
View File
@@ -669,7 +669,9 @@ def install_rocm_zluda():
os.environ.setdefault('TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL', '1')
if args.use_nightly:
if rocm.version is None or float(rocm.version) >= 6.3: # assume the latest if version check fails
if rocm.version is None or float(rocm.version) >= 6.4: # assume the latest if version check fails
torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4')
elif rocm.version == "6.3":
torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.3')
else: # oldest rocm version on nightly is 6.2.4
torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.2.4')