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

[Hewentao123]

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?

[piaozhx]

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

[Hewentao123]

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！