diff --git a/scripts/check_modules.py b/scripts/check_modules.py
index 140b7b4c..57cbe0fa 100644
--- a/scripts/check_modules.py
+++ b/scripts/check_modules.py
@@ -94,15 +94,15 @@ def install(module_name: str, module_version: str):
         if module_name == "torch":
             if "Navi 3" in amd_gpus:
                 # 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"
+                module_version = "2.4.1+rocm6.1"
+                index_url = "https://download.pytorch.org/whl"
             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 5.5+
-                module_version = "0.16.0+rocm5.5"
-                index_url = "https://download.pytorch.org/whl/rocm6.2"
+                module_version = "0.20.0+rocm6.1"
+                index_url = "https://download.pytorch.org/whl"
             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/rocm6.2"
+        index_url = "https://download.pytorch.org/whl"
 
     return module_version, index_url