Skip to content

Conversation

w-e-w
Copy link
Collaborator

@w-e-w w-e-w commented Apr 29, 2025

Description

Upgread to PyTorch==2.7.0 and xFormers==0.0.30
support using Nvidia 50 series Blackwell GPUs using official wheels

didn't found issues during testing
note that testing is done on a 3090 as I don't have a 50 series card so I cant say for sure
I prefer other people to also verify this if possible

base comment below:
seem to work on (I assume on windows) but might have some issue on linux

Checklist:

Deprecated early_access_blackwell_wheels()
@w-e-w
Copy link
Collaborator Author

w-e-w commented Apr 29, 2025

@catboxanon

@semolina5
Copy link

semolina5 commented Apr 30, 2025

I tried your patches. My setup:

  • Debian 12 x86_64, Linux kernel 6.14.3
  • NVIDIA drivers 570.144
  • Geforce RTX 5060 Ti with 16 GB RAM

I installed stable-diffusion-webui as shown in the documentation section. After starting webui.sh, I received the error message

NVIDIA Graphics Device with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.

After that, I ran the commands below:

git checkout 5c0fc36       # 85 commits ahead of master
webui.sh --reinstall-torch
webui.sh --reinstall-xformers
webui.sh

webui.sh works now and I am able to generate images. xformers still doesn't run, but I don't care about that.

Many thanks for your patches!

@uyheangte
Copy link

Description

Upgread to PyTorch==2.7.0 and xFormers==0.0.30 support using Nvidia 50 series Blackwell GPUs using official wheels

didn't found issues during testing but I prefer other people to also verify this if possible

Checklist:

I'm running on RTX 5090, and I can confirmed that this works! Really thank you for your contribution.

@catboxanon catboxanon merged commit 3064b21 into dev May 1, 2025
6 checks passed
@catboxanon catboxanon deleted the Pytorch==2.7.0-xFormers==0.0.30 branch May 1, 2025 16:57
w-e-w added a commit to w-e-w/stable-diffusion-webui that referenced this pull request May 2, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants