Giters

svip-lab / impersonator

PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis

Home Page:https://svip-lab.github.io/project/impersonator

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

I run it on win10 successfully, but it always use cpu not GPU

Hewentao123 opened this issue · comments

commented

On my virtual environment , I used the code to test CUDA-GPU successfully and I set the gpu-ids=0, but it still use cpu. I don't know the reason. could you help me?

commented

that's impossible because we do not have cpu version. if you run the code successfully, it will definitely use gpu.

commented

Thanks for your revision. Maybe, I need to set the availiable num_workers and increase the batsize as the GPU has been in idle waiting state. Thanks！