Colab 스테이블디퓨전 xFormers can't load C++/CUDA extensions. xFormers was built for 에러 해결방법

Colab 스테이블디퓨전 xFormers can't load C++/CUDA extensions 에러 해결방법

스테이블디퓨전 colab xformer 관련 에러

 

최근 Colab 스테이블디퓨전 web-ui 노트북을 실행시키면 아래와 같은 에러가 발생하는 것을 볼 수 있습니다.

에러 메시지

WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for:
PyTorch 2.0.1+cu118 with CUDA 1108 (you have 2.1.0+cu118)
Python 3.10.11 (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

해당 에러가 발생 후 종료가 되는 건 아니고, gradio도 접속이 잘 되지만 프롬프트를 입력하여 이미지 생성 시 에러가 출력되면서 동작이 되지 않는 현상이 있습니다.

 

해결방법

해당 에러는 현재 github에 이슈로 등록되어 있는 상태이고, 임시 해결책이 나와 있습니다.

아래와 같이 맨 처음 블럭에 코드를 추가하면 되는데요. 

!pip install lmdb
!pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 torchtext==0.15.2+cpu torchdata==0.6.1 --index-url https://download.pytorch.org/whl/cu118

이런 식으로 추가를 하시면 해당 에러를 고칠 수 있습니다.

xformer 에러 해결 방법

모두 즐거운 이미지 생성하시기 바랍니다. 

관련 github 이슈링크는 아래를 참고 하세요.

https://github.com/TheLastBen/fast-stable-diffusion/issues/2615

Google Collab Pro: WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions · Issue #2615 · TheLastBen/fast-stable-diffusion