Cuda runtime error (801) multiple workers

Hi,

I am getting the same error and I am using set_start_method(‘spawn’).

Could you please help a check what is wrong with this code.