Use nightly/cu128 for nvidia 50xx

Author: cmdr2

tests/test_platform_detection.py

@@ -120,7 +120,7 @@ def test_nvidia_5xxx_gpu_windows(monkeypatch):
     monkeypatch.setattr("torchruntime.platform_detection.os_name", "Windows")
     monkeypatch.setattr("torchruntime.platform_detection.arch", "amd64")
     gpu_infos = [GPU(NVIDIA, "NVIDIA", "2c02", "GB203 [GeForce RTX 5080]", True)]
-    assert get_torch_platform(gpu_infos) == "nightly/cu124"
+    assert get_torch_platform(gpu_infos) == "nightly/cu128"
 
 
 def test_intel_gpu_windows(monkeypatch):
@@ -143,7 +143,7 @@ def test_nvidia_5xxx_gpu_linux(monkeypatch):
     monkeypatch.setattr("torchruntime.platform_detection.os_name", "Linux")
     monkeypatch.setattr("torchruntime.platform_detection.arch", "x86_64")
     gpu_infos = [GPU(NVIDIA, "NVIDIA", "2c02", "GB203 [GeForce RTX 5080]", True)]
-    assert get_torch_platform(gpu_infos) == "nightly/cu124"
+    assert get_torch_platform(gpu_infos) == "nightly/cu128"
 
 
 def test_intel_gpu_mac(monkeypatch):

torchruntime/platform_detection.py

@@ -111,7 +111,7 @@ def _get_platform_for_discrete(gpu_infos):
         if os_name in ("Windows", "Linux"):
             device_names = set(gpu.device_name for gpu in gpu_infos)
             if any(BLACKWELL_DEVICES.search(device_name) for device_name in device_names):
-                return "nightly/cu124"
+                return "nightly/cu128"
 
             return "cu124"
         elif os_name == "Darwin":