I run it on win10 successfully, but it always use cpu not GPU
Hewentao123 opened this issue · comments
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?
that's impossible because we do not have cpu version. if you run the code successfully, it will definitely use gpu.
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!