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

ROCm 6.0 update mpmath to 1.3.0

Browse Source
This commit is contained in:
Disty0
2024-03-25 00:15:24 +03:00
parent c953dfdfd7
commit c4668a79fa
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+2 -2
View File
@@ -526,8 +526,8 @@ def check_torch():
torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision')
else:
if rocm_ver == "6.0":
# mpmath 1.3.0 crashes at import
torch_command = os.environ.get('TORCH_COMMAND', f'torch torchvision mpmath==1.2.1 --pre --index-url https://download.pytorch.org/whl/nightly/rocm{rocm_ver}')
# mpmath 1.4.0a0 crashes at import
torch_command = os.environ.get('TORCH_COMMAND', f'torch torchvision mpmath==1.3.0 --pre --index-url https://download.pytorch.org/whl/nightly/rocm{rocm_ver}')
elif rocm_ver == "5.7":
torch_command = os.environ.get('TORCH_COMMAND', f'torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/rocm{rocm_ver}')
elif rocm_ver in {"5.5", "5.6"}: