Hi,
Thank you. But is there any way to globally force to disable cuda? This will make sure the training can only run on cpu. The test will be more independent.
Otherwise if we use a flag, the code may still run gpu somewhere inside by mistake.