From ee86eedf0145ed11fb6359ba99277f5b8ab8251e Mon Sep 17 00:00:00 2001 From: creachec <46060787+creachec@users.noreply.github.com> Date: Fri, 31 Mar 2023 17:22:00 -0300 Subject: [PATCH] Right link on pytorch installer for linux rocm Right link on pytorch installer for linux rocm --- installer/lib/installer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/installer/lib/installer.py b/installer/lib/installer.py index 8ab512eee8..42079cd90c 100644 --- a/installer/lib/installer.py +++ b/installer/lib/installer.py @@ -456,7 +456,7 @@ def get_torch_source() -> (Union[str, None],str): optional_modules = None if OS == "Linux": if device == "rocm": - url = "https://download.pytorch.org/whl/rocm5.2" + url = "https://download.pytorch.org/whl/rocm5.4.2" elif device == "cpu": url = "https://download.pytorch.org/whl/cpu"