RuntimeError: arguments are located on different GPUs

sergeytulyakov / mocogan

MoCoGAN: Decomposing Motion and Content for Video Generation

573 stars 113 forks source link

RuntimeError: arguments are located on different GPUs #34

Open cyhhhhhhit opened 3 years ago

cyhhhhhhit commented 3 years ago

Same issue #30 Is there any new progress about this question?

Fanny-Yuan commented 3 years ago

Hi, this happened because pytorch uses number 0 GPU by default but number 0 GPU was used by another user at that time. When number 0 GPU is available this problem won't happen. However, if you want to run it at other GPUs, you may need to refactor the code or try another pytorch version MoCoGAN code https://github.com/DLHacks/mocogan.