Skip to content

Commit

Permalink
fix(installer): pytorch and ROCm versions are incompatible
Browse files Browse the repository at this point in the history
Each version of torch is only available for specific versions of CUDA and ROCm.
The Invoke installer tries to install torch 2.4.1 with ROCm 5.6 support, which
does not exist. As a result, the installation falls back to the default CUDA
version so AMD GPUs aren't detected. This commits fixes that by bumping the
ROCm version to 6.1, as suggested by the PyTorch documentation. [1]

The specified CUDA version of 12.4 is still correct according to [1] so it does
need to be changed.

Closes #7146

[1]: https://pytorch.org/get-started/previous-versions/#v241
  • Loading branch information
max-maag committed Oct 18, 2024
1 parent 5440c03 commit 9b9632e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion installer/lib/installer.py
Original file line number Diff line number Diff line change
Expand Up @@ -407,7 +407,7 @@ def get_torch_source() -> Tuple[str | None, str | None]:
optional_modules: str | None = None
if OS == "Linux":
if device == GpuType.ROCM:
url = "https://download.pytorch.org/whl/rocm5.6"
url = "https://download.pytorch.org/whl/rocm6.1"
elif device == GpuType.CPU:
url = "https://download.pytorch.org/whl/cpu"
elif device == GpuType.CUDA:
Expand Down

0 comments on commit 9b9632e

Please sign in to comment.