From 8544f3af4b9d8a158eb821bc021b767f11e0b53b Mon Sep 17 00:00:00 2001
From: Kavya Mali <kavyamali2904@gmail.com>
Date: Fri, 7 Nov 2025 15:03:37 +0530
Subject: [PATCH] Supported Pytorch added for Maxwell.

---
 modules/launch_utils.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/modules/launch_utils.py b/modules/launch_utils.py
index 394804f67..3280aae87 100644
--- a/modules/launch_utils.py
+++ b/modules/launch_utils.py
@@ -347,9 +347,19 @@ def early_access_blackwell_wheels():
         return f'pip install {ea_whl.get(sys.version_info.minor)}'
 
 
+def maxwell():
+    """For Maxwell GPUs"""
+    cc = get_cuda_comp_cap()
+    if 5.0 <= cc < 6.0:  # Maxwell, Compute Capability 5.0/5.2
+        torch_command = os.environ.get('TORCH_COMMAND', f"pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126")
+        return torch_command
+        
+    return None
+
+
 def prepare_environment():
     torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu128")
-    torch_command = os.environ.get('TORCH_COMMAND', f"pip install torch==2.7.0 torchvision==0.22.0 --extra-index-url {torch_index_url}")
+    torch_command = maxwell() or os.environ.get('TORCH_COMMAND', f"pip install torch==2.7.0 torchvision==0.22.0 --extra-index-url {torch_index_url}")
     if args.use_ipex:
         if platform.system() == "Windows":
             # The "Nuullll/intel-extension-for-pytorch" wheels were built from IPEX source for Intel Arc GPU: https://github.com/intel/intel-extension-for-pytorch/tree/xpu-main
@@ -512,3 +522,4 @@ def dump_sysinfo():
         file.write(text)
 
     return filename
+