The error is raised, since you are most likely using multiprocessing in the DataLoader by specifying num_workers>0, which would then try to re-initialize the CUDA context.
Did you try the suggested workaround using the 'spawn' start method?