RuntimeError: Caught RuntimeError in DataLoader worker process 0.

RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method