Как связаны версии трёх вещей — pyTorch, CUDA и CUDA compute capability?

Question

[SmeliyR]

Поставил последний ComfyUI, но он послал меня куда подальше.

NVIDIA P106-100 with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA P106-100 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

Начал разбираться и уже пару часов не могу понять, как между собой связаны версии трёх вещей - pyTorch, CUDA и CUDA compute capability?
pyTorch привязывается к версии CUDA или к версии compute capability?
или и к тому и к тому? или версии CUDA или к версии compute capabilityв принципе связаны между собой?
как на сайте pytorch выбрать нужную compute compability ведь там нет такого выбора!

Answer 1

[freeExec]

Связь такая pyTorch -> CUDA -> GPU [CUDA compute capability]
Т.е. твоя железка ограничивает максимально доступную версию CUDA. А соответственно для этой версии CUDA когда-то был свой последний pyTorch.

Но судя по это статье твой sm_61 вполне ещё вчера поддерживался.

Comments

[SmeliyR]

Т.е. твоя железка ограничивает максимально доступную версию CUDA.

на данный момент поставил дрова с CUDA 13.
Вроде это самая новая. Так что если и органичивает, то непонятно как.

Sat Sep  6 21:01:18 2025
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 580.88                 Driver Version: 580.88         CUDA Version: 13.0     |
+-----------------------------------------+------------------------+----------------------+
| GPU  Name                  Driver-Model | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M.

Статью по ссылке глянул, в принципе, это то что надо - но там инфа старая. И неофициальная - какой то юзер с форума вручную выковыривал, как я понимаю.

Решил попробовать перебором. Первая же попытка - torch 2.6.0+cu126 сработала, и нормально ужилась с sm_61

Но метод научного тыка - это всё-таки не научно. Если кто знает где официально есть такая таблица совместимости - киньте ссылкой...

[freeExec]

SmeliyR, Ограничивает тем, что поставляет компилятор кода cuda под архитектуру sm_xx
Да, т.е. тут кроме самого куда, автор программы должен собрать код под нужную архитектуру.