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

[Anonymous]

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

ОС: Windows 11

Графический процессор: NVIDIA GeForce RTX 5090

Версия драйвера: 581.80

Вывод nvidia-smi:

Версия драйвера NVIDIA-SMI 581.80: 581.80 Версия CUDA: 13.0

| 0 NVIDIA GeForce RTX 5090 WDDM | 00000000:01:00.0 Включено |

| 0% 36C P8 14Вт/600Вт | 1094 МБ / 32607 МБ |

Версия Python: 3.12
Версия PyTorch:

Версия PyTorch: 2.7.0.dev20250310+cu124

CUDA доступно: True

Количество устройств CUDA: 1

Устройство 0: NVIDIA GeForce RTX 5090

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

Предупреждение пользователя:

NVIDIA GeForce RTX 5090 с возможностью CUDA sm_120 несовместима с текущей версией Установка PyTorch.

Текущая установка PyTorch поддерживает возможности CUDA sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90.

Ошибка при выполнении простой операции графического процессора:

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

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

Результат:

RuntimeError: ошибка CUDA: образ ядра недоступен для выполнения на устройстве

Что я хочу знать:
Предоставляет ли 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