AI 이미지/ComfyUI

NVidia 50 시리즈와 ComfyUI

하늘이푸른오늘 2025. 4. 24. 14:55

NVidia 50시리즈가 여러가지로 문제를 일으키고 있나봅니다. 저도 5070ti를 주문해놓고 낼모레 배송을 기다리고 있는데, ComfyUI가 Blackwell 아키텍처에서 문제가 있다는 모양입니다.

업데이트:

그냥 아래 링크에서 다운로드 받아 설치하면 된다고 하네요.  안전화버전의 PyTorch 2.7 cu128 버전에서 돌아가는 ComfyUI입니다.

https://github.com/comfyanonymous/ComfyUI/releases/download/latest/new_ComfyUI_windows_portable_nvidia_cu128_or_cpu.7z

업데이트:

이제는 더 이상 신경쓸 필요 없습니다!!

문제의 원인

ComfyUI를 설치하고 실행시키면 로그가 나오는데 문제는 Pytorch 버전때문입니다.

  • ComfyUI의 Python 버전 3.12.8 / Pytorch 버전 2.6.0+cu126(CUDA 12.6)

문제는 Pytorch 버전 2.6.0+cu126이 Blackwell 아키텍처를 지원하지 못해서 ComfyUI가 오류가 발생하는 거고, 이를 해결하기 위해서는 안정화버전이 아닌 nightly 버전인 PyTorch 2.7 nightly with CUDA 12.8 을 설치해야 하는 겁니다.

Pytorch 업데이트

가장 간단한 방법은 아래의 명령을 실행해보는 것입니다.

pip install --force-reinstall --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

제 생각에 이 명령으로 거의 99% 해결될 거라고 생각합니다. 그런데... 그래도 안된다고 하면 아래 절차에 따라 삭제후 다시 설치해 볼 수 있습니다.

  • Comfy\ComfyUI_windows_portable 폴더에 들어갑니다.
  • 다음 명령을 실행시킵니다. 원래 ComfyUI에 포함된 python을 사용해 현재의 Python 을 삭제하는 명령입니다.
.\python_embeded\python.exe -m pip uninstall torch torchvision torchaudio -y
  • PyTorch 2.7 Nightly with CUDA 12.8 을 설치합니다.
.\python_embeded\python.exe -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
  • PyTorch 2.7이 정상적으로 설치되었는지 확인합니다.
.\python_embeded\python.exe -c "import torch; print(torch.version); print(torch.cuda.is_available()); print(torch.cuda.get_device_name(0))"

이렇게 실행했을 때 아래와 같은 내용이 나와야 합니다. 

The PyTorch version (e.g., 2.7.0.dev20250411+cu128).
True for CUDA availability.
NVIDIA GeForce RTX 5070 as the device name.
  • 만약 CUDA가 문제가 있다면 NVidia 드라이버가 CUDA 12.8 번전을 지원하는지 확인해보고( run nvidia-smi to check; it should show CUDA Version: 12.8 ???) 드라이버를 최신버전(572.70 이상) 으로 설치해야 합니다.

이 정도면 대략 다 해결될 거라고 생각하는데, 만약 그래도 안된다면 ComfyUI를 완전히 새로 깔고 시작하는 게 좋을 것 같습니다.

그래도 모르시겠다면 https://github.com/comfyanonymous/ComfyUI/discussions/6643 를 참고해보세요.

민, 푸른하늘

이상입니다.