search

open-mmlab / PIA

[CVPR 2024] PIA, your Personalized Image Animator. Animate your images by text prompt, combing with Dreambooth, achieving stunning videos. PIA，你的个性化图像动画生成器，利用文本提示将图像变为奇妙的动画
https://pi-animator.github.io/
Apache License 2.0
913 stars 75 forks source link

xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.1.2+cu121 with CUDA 1201 (you have 2.1.2+cpu) #30

Closed guahunyo closed 10 months ago

guahunyo commented 10 months ago

(pia) PS E:\PIA> python inference.py --config=example/config/majic_girl.yaml WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.1.2+cu121 with CUDA 1201 (you have 2.1.2+cpu)

是我的xformers版本问题吗 还是cuda版本问题 如果需要降版本的话要降到哪个版本 我的CUDA是12.2.79 python: 3.10.8   xformers: 0.0.21

ymzhang0319 commented 10 months ago

你好@guahunyo，似乎是因为你的pytorch是cpu的。 下载一个cuda版本的pytorch也许可以解决。可以参考官网，下载一个相应版本的：https://pytorch.org/get-started/locally/