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

RuntimeError: GET was unable to find an engine to execute this computation #49

Closed mamianshusheng closed 4 months ago

mamianshusheng commented 4 months ago

RuntimeError: GET was unable to find an engine to execute this computation

Something wrong will happen like the above unless I use torch == 2.0.0(I used torch==2.3.1 following your pia.yml) . I hope you can change the related contents in your pia.yml

zengyh1900 commented 4 months ago

Hi @mamianshusheng, I appreciate your interest. I think you may have installed the CPU version. You can install torch following the official guide: https://pytorch.org/

mamianshusheng commented 4 months ago

Hi @mamianshusheng, I appreciate your interest. I think you may have installed the CPU version. You can install torch following the official guide: https://pytorch.org/

Well，I truly installed the GPU version. I used this command: pip install torch==2.3.0+cu118 torchvision==0.18.0+cu118 torchaudio==2.3.0+cu118 --index-url https://download.pytorch.org/whl/cu118

I think torch=2.3.0 is too new to fit the environment of the project