search

DinoMan / speech-driven-animation

949 stars 289 forks source link

How to run on CPU #38

Closed shrnjad closed 4 years ago

shrnjad commented 4 years ago

I do not have a CUDA device, when running the example, I get this error:

AttributeError: module 'torch._C' has no attribute '_cuda_getDevice'

How can I run the network on the CPU? What is the syntax?

DinoMan commented 4 years ago

The default argument for the anmator is gpu=-1 so if you do:

va = sda.VideoAnimator() instead of: va = sda.VideoAnimator(gpu=0)

it will use the CPU.

shrnjad commented 4 years ago

Thank you!