[GH-ISSUE #251] Update Colab notebook for CUDA 12.1 #134

Closed
opened 2026-10-27 21:13:36 +00:00 by bsakaguchi · 4 comments
Owner

Originally created by @BlakeOne on GitHub (Dec 19, 2023).
Original GitHub issue: https://github.com/Comfy-Org/ComfyUI-Manager/issues/251

Now that CUDA 12.1 is available it seems the Colab notebook needs an update... Here is the warning message which can be reproduced by running the notebook on Colab.

WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for:
PyTorch 2.1.2+cu118 with CUDA 1108 (you have 2.1.2+cu121)
Python 3.10.13 (you have 3.10.12)
Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers)
Memory-efficient attention, SwiGLU, sparse and more won't be available.
Set XFORMERS_MORE_DETAILS=1 for more details

Originally created by @BlakeOne on GitHub (Dec 19, 2023). Original GitHub issue: https://github.com/Comfy-Org/ComfyUI-Manager/issues/251 Now that CUDA 12.1 is available it seems the Colab notebook needs an update... Here is the warning message which can be reproduced by running the notebook on Colab. WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.1.2+cu118 with CUDA 1108 (you have 2.1.2+cu121) Python 3.10.13 (you have 3.10.12) Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers) Memory-efficient attention, SwiGLU, sparse and more won't be available. Set XFORMERS_MORE_DETAILS=1 for more details
Author
Owner

@ghost commented on GitHub (Dec 20, 2023):

Just change in your notebook the line from:

!pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

to:

!pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
<!-- gh-comment-id:1864964715 --> @ghost commented on GitHub (Dec 20, 2023): Just change in your notebook the line from: ``` !pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 ``` to: ``` !pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 ```
Author
Owner

@ltdrdata commented on GitHub (Dec 21, 2023):

Just change in your notebook the line from:

!pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

to:

!pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

Is it work? I can't test now.

<!-- gh-comment-id:1865410662 --> @ltdrdata commented on GitHub (Dec 21, 2023): > Just change in your notebook the line from: > > ``` > !pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 > ``` > > to: > > ``` > !pip3 install xformers!=0.0.18 torch==2.1.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 > ``` Is it work? I can't test now.
Author
Owner

@BlakeOne commented on GitHub (Dec 21, 2023):

Yep it fixed it, thanks! :)

<!-- gh-comment-id:1865443266 --> @BlakeOne commented on GitHub (Dec 21, 2023): Yep it fixed it, thanks! :)
Author
Owner

@ltdrdata commented on GitHub (Dec 21, 2023):

I just updated this.

<!-- gh-comment-id:1866125668 --> @ltdrdata commented on GitHub (Dec 21, 2023): I just updated this.
Sign in to join this conversation.
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
Comfy-Org/ComfyUI-Manager#134
No description provided.