PyTorch erkennt RTX 5090 (sm_120) unter Windows 11 nicht – CUDA-Fehler: Kein Kernel-Image verfügbar ⇐ Python

[Anonymous]

Ich versuche, PyTorch mit einer NVIDIA GeForce RTX 5090 (Blackwell-Architektur, CUDA Compute Capability sm_120) unter Windows 11 zu verwenden, und es treten immer wieder Kompatibilitätsprobleme auf. PyTorch erkennt CUDA, aber jeder GPU-Vorgang schlägt mit einem Kernel-Image-Fehler fehl.
Systemdetails:

Betriebssystem: Windows 11

GPU: NVIDIA GeForce RTX 5090

Treiberversion: 581.80

nvidia-smi Ausgabe:

NVIDIA-SMI 581.80 Treiberversion: 581.80 CUDA-Version: 13.0

| 0 NVIDIA GeForce RTX 5090 WDDM | 00000000:01:00.0 Am |

| 0% 36C P8 14W / 600W | 1094MiB / 32607MiB |

Python-Version: 3.12
PyTorch-Version:

PyTorch-Version: 2.7.0.dev20250310+cu124

CUDA verfügbar: Richtig

CUDA-Geräteanzahl: 1

Gerät 0: NVIDIA GeForce RTX 5090

Warnmeldung:

Benutzerwarnung:

NVIDIA GeForce RTX 5090 mit CUDA-Fähigkeit sm_120 ist nicht mit der aktuellen PyTorch-Installation kompatibel.

Die aktuelle PyTorch-Installation unterstützt CUDA-Funktionen sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90.

Fehler beim Ausführen einer einfachen GPU-Operation:

Code: Select all

x = torch.randn(1000, 1000, device="cuda")

Ergebnis:

RuntimeError: CUDA-Fehler: Für die Ausführung auf dem Gerät ist kein Kernel-Image verfügbar

Was ich wissen möchte:
Stellt PyTorch derzeit offizielle Windows-Binärdateien bereit, die die in der verwendete sm_120-Architektur unterstützen? RTX 5090?
Wenn nicht, sind dies die einzigen aktuellen Optionen?

• Linux/WSL2 verwenden und jede Nacht das neueste PyTorch mit CUDA 12.8+ installieren
• PyTorch aus dem Quellcode unter Windows erstellen mit:

  Code: Select all

  set TORCH_CUDA_ARCH_LIST=8.6;9.0;12.0

Hat jemand RTX 5090 + Windows 11 + PyTorch erfolgreich verwendet?
Wenn ja, teilen Sie uns bitte die genaue Arbeitskombination mit:
PyTorch-Version
CUDA-Version
Installationsbefehl
Ich versuche zu verstehen den aktuellen Supportstatus für Blackwell-GPUs unter Windows.