extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-04 12:40:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update the index url for AMD ROCm torch install

Browse Source
This commit is contained in:
cmdr2 2025-01-06 19:48:44 +05:30
parent 493035df16
commit 52aaef5e39
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/check_modules.py
View File

@ -93,16 +93,16 @@ def install(module_name: str, module_version: str):
amd_gpus = setup_amd_environment()
if module_name == "torch":
if "Navi 3" in amd_gpus:
# No AMD 7x00 support in rocm 5.2, needs nightly 5.5. build
module_version = "2.1.0.dev-20230614+rocm5.5"
index_url = "https://download.pytorch.org/whl/nightly/rocm5.5"
# No AMD 7x00 support in rocm 5.2, needs 5.5+
module_version = "2.1.0+rocm5.5"
index_url = "https://download.pytorch.org/whl/rocm6.2"
else:
module_version = "1.13.1+rocm5.2"
elif module_name == "torchvision":
if "Navi 3" in amd_gpus:
# No AMD 7x00 support in rocm 5.2, needs nightly 5.5. build
module_version = "0.16.0.dev-20230614+rocm5.5"
index_url = "https://download.pytorch.org/whl/nightly/rocm5.5"
# No AMD 7x00 support in rocm 5.2, needs 5.5+
module_version = "0.16.0+rocm5.5"
index_url = "https://download.pytorch.org/whl/rocm6.2"
else:
module_version = "0.14.1+rocm5.2"
elif os_name == "Darwin":
@ -286,7 +286,7 @@ def apply_torch_install_overrides(module_version: str):
module_version += "+cu117"
index_url = "https://download.pytorch.org/whl/cu117"
elif is_amd_on_linux():
index_url = "https://download.pytorch.org/whl/rocm5.2"
index_url = "https://download.pytorch.org/whl/rocm6.2"
return module_version, index_url