Я люблю стабильный софт. Особенно, когда дело касается важных задач и повседневной работы. Но при работе с нейросетями, даже немного устаревшая библиотека может работать куда медленнее актуальной версии. Поэтому небольшая инструкция на тему как обновить самое необходимое.
Вкусности от NVIDIA
Сперва проверяем что именно поддерживается, а для этого вбиваем команду nvidia-smi в консоли.
CUDA
Переходим по ссылке и качаем самую свеженькую версию CUDA для нашей системы. В моём случае это 12.6
Запускаем инсталлятор и ставим необходимые рантаймы и компилятор. После этого, через PowerShell проверяем есть ли cuda в системном PATH и если нет, добавляем.
$env:path -split ';' | ForEach-Object { $_ }
Если следующая команда не выдаёт ошибку, всё хорошо, идём дальше.
nvcc -V
cuDNN
Всё аналогично предыдущему пункту. Качаем отсюда, ставим, проверяем чтобы в PATH был путь.
C:\Program Files\NVIDIA\CUDNN\v9.6\bin
Питонские штучки
PyTorch
Идём на https://pytorch.org/, выбираем Stable -> Windows -> Pip -> Python -> CUDA 12.x и получаем следующую команду:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
Если её ввести просто так, то эти пакеты будут установлены в систему, а ComfyUI, как и многие другие AI штуки имеют свою внутреннюю версию Python. Поэтому переходим в директорию с ComfyUI, заходим в папку python_embeded и пишем в адресной строке cmd. Чуточку меняем команду, чтобы получилось следующее и запускаем.
python.exe -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
xformers
Не закрывая консоль, вводим ещё одну команду.
python.exe -m pip install -U xformers==0.0.28.post3 --index-url https://download.pytorch.org/whl/cu124
Тут стоит пояснить, что ставится версия 28, так как на момент написания данной инструкции версия 29 могла выдавать ошибки.
Готово. Осталось только запустить ComfyUI и проверить чтобы в консоли были актуальные версии:
pytorch version: 2.5.1+cu124
xformers version: 0.0.28.post3