PyTorch не распознает RTX 5090 (sm_120) в Windows 11 — ошибка CUDA: образ ядра недоступен ⇐ Python

[Anonymous]

Я пытаюсь использовать PyTorch с NVIDIA GeForce RTX 5090 (архитектура Blackwell, вычислительные возможности CUDA sm_120) в Windows 11, и постоянно сталкиваюсь с проблемами совместимости. PyTorch обнаруживает CUDA, но любая операция графического процессора завершается с ошибкой образа ядра.
Сведения о системе:

Код: Выделить всё

OS: Windows 11

GPU: NVIDIA GeForce RTX 5090

Driver version: 581.80

Вывод nvidia-smi:

Код: Выделить всё

NVIDIA-SMI 581.80                 Driver Version: 581.80         CUDA Version: 13.0
|   0  NVIDIA GeForce RTX 5090      WDDM  |   00000000:01:00.0  On |
|  0%   36C    P8             14W /  600W |    1094MiB /  32607MiB |

Python version: 3.12

Версия PyTorch:

Код: Выделить всё

PyTorch version: 2.7.0.dev20250310+cu124
CUDA available: True
CUDA device count: 1
Device 0: NVIDIA GeForce RTX 5090

Предупреждающее сообщение:

Код: Выделить всё

UserWarning:
NVIDIA GeForce RTX 5090 with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90.

Error when running a simple GPU operation:
x = torch.randn(1000, 1000, device="cuda")

Результат:

Код: Выделить всё

RuntimeError: CUDA error: no kernel image is available for execution on the device

Что я хочу знать:
Предоставляет ли PyTorch в настоящее время какие-либо официальные двоичные файлы Windows, поддерживающие архитектуру sm_120, используемую в RTX 5090?
Если нет, то являются ли это единственными текущими вариантами?
Использование Linux/WSL2 и ежевечерняя установка последней версии PyTorch с CUDA 12.8+
Сборка PyTorch из исходного кода в Windows с помощью:

Код: Выделить всё

set TORCH_CUDA_ARCH_LIST=8.6;9.0;12.0

Кто-нибудь успешно использовал RTX 5090 + Windows 11 + PyTorch?
Если да, поделитесь точной рабочей комбинацией:
Версия PyTorch
Версия CUDA
Команда установки
Я пытаюсь понять текущий статус поддержки графических процессоров Blackwell на Windows.

Подробнее здесь: https://stackoverflow.com/questions/798 ... -no-kernel