Your error seems to be this one 5.



RuntimeError: cuda runtime error (71) : operation not supported at 


https://pytorch.org/docs/stable/notes/windows.html#multiprocessing-error-without-if-clause-protection