Nvidia 50 Series (Blackwell) support thread: How to get ComfyUI running on your new 50 series GPU. #6643
Replies: 16 comments 47 replies
-
There are also Windows prebuilt wheels of pytorch on cuda 128 that nvidia gave w-e-w to publish https://huggingface.co/w-e-w/torch-2.6.0-cu128.nv |
Beta Was this translation helpful? Give feedback.
-
The Windows version said: |
Beta Was this translation helpful? Give feedback.
-
nf4 node does not work. |
Beta Was this translation helpful? Give feedback.
-
ImportError: tokenizers>=0.21,<0.22 is required for a normal functioning of this module, but found tokenizers==0.20.3. How can I do to fix problem? |
Beta Was this translation helpful? Give feedback.
-
Sorry - I have been banging my head against this for 6 hours - trying to run a cosmos workflow with 5090 - without SageAttention or TorchCompile - a 14 minute job on 4090 is taking 25 minutes. To turn them on I installed Triton and Sageattention - then I get this error afterwards whether I bypass the patch and compile or not:- KSampler I have run this through GPT and followed every instruction. Step 1: Verify CUDA Toolkit Installation python If it does not match 12.8, you may need to reinstall PyTorch with the correct CUDA version. ✅ Step 2: Check NVIDIA Toolkit and Drivers Update your NVIDIA driver (Download latest) ✅ Step 3: Fix Microsoft Visual Studio Build Tools Run: sh |
Beta Was this translation helpful? Give feedback.
-
anyone knows if other pytorch's cuda versions like 12.6 will work with blackwell? |
Beta Was this translation helpful? Give feedback.
-
When I use [ComfyUI package with a cuda 12.8 torch build], many custom_nodes are "IMPORT FAILED" including Manager, InstantID, ReActor ... |
Beta Was this translation helpful? Give feedback.
-
Is is possible to build a pytorch that works for 5090? If so how to do it? |
Beta Was this translation helpful? Give feedback.
-
Having given up on Portable for now - too many errors lol - I am using WSL - Ubuntu - everything else is set up and working in Comfyui - I have the latest pytorch nightly from pip install --pre torch torchvision torchaudio --index-url[ https://download.pytorch.org/whl/nightly/cu128] - but when I use Sageattention I get this: |
Beta Was this translation helpful? Give feedback.
-
im using comfyui over pinokio, is there any way to update my comfy to work with my new 5080? i deleted the old files and changed in the torch.js on the patch above, but it wont start |
Beta Was this translation helpful? Give feedback.
-
Will using a docker allow for a working torchvision in windows? |
Beta Was this translation helpful? Give feedback.
-
Can someone write a little guide for getting a docker running torchvision, etc on window with the portable Blackwell comfyui release? Please write it for a normal user who has no programming knowledge. There are basic instructions in the OP, but what even is a docker? "docker run -p 8188:8188 --gpus all -it --rm nvcr.io/nvidia/pytorch:25.01-py3" Where is this command supposed to be run? Is a docker something to be installed to system python or standalone folder? How would this be installed on a fresh portable comfyui install? With more Blackwell cards trickling out, there will most likely be more users needing help setting this up. Thank you. |
Beta Was this translation helpful? Give feedback.
-
To anyone having trouble with the portable version for Blackwell gpus, do not update it! I noticed during updating that it uninstalled the cu128 and installed another version for example. Torchvision works with a fresh install without updating! |
Beta Was this translation helpful? Give feedback.
-
Excelente |
Beta Was this translation helpful? Give feedback.
-
Are there no windows pytorch options available for cuda 12.8 still? Any help would be greatly appreciated. |
Beta Was this translation helpful? Give feedback.
-
For those who want to use Sage Attention + Hunyuan + RTX 50xx with a simple tutorial, it is interesting to keep the files inside WSL instead of using the files in Windows due to the reading time (you can test it and you will understand) that's why I recommend copying your comfyui to WSL. https://github.com/alisson-anjos/ComfyUI_Tutoriais/blob/main/WSL/install.md Notes: It is necessary to use the kijai patch node for sage attention because it currently only works with native sage attention and not varlen, at the end of the README there is a workflow with this node. And for those who want to compile SageAttention due to some lack of compatibility with the OS, it is necessary to compile Triton and then Sage Attention, to compile Sage Attention it is necessary to change setup.py with the code in this issue: [Sage Attention Issue] (thu-ml/SageAttention#107 (comment)) Thanks to Kijai for precompiling triton and sage attention. |
Beta Was this translation helpful? Give feedback.
-
I will try keeping this post up to date as much as possible with the latest developments.
To get your nvidia 50 series GPU working with ComfyUI you need a pytorch that has been built against cuda 12.8
In the next few months there will likely be a lot of performance improvements landing in pytorch for these GPUs so I recommend coming back to this page and updating frequently.
Windows
Windows users can download this standalone ComfyUI package with a cuda 12.8 torch build
Manual Install
At this moment Blackwell is not yet supported by stable pytorch.
pytorch nightly cu128 is available for Linux:
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
You can also use the Nvidia Pytorch Docker Container as an alternative which might give more performance.
Link: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch
Here's how to use it:
docker run -p 8188:8188 --gpus all -it --rm nvcr.io/nvidia/pytorch:25.01-py3
Inside the docker container:
Beta Was this translation helpful? Give feedback.
All reactions