- use torch.spawn
- dataloader: pin_memory=True.
- dataloader: num_worker>1.
ERROR:RuntimeError: CUDA driver error: initialization error
I guess you are trying to re-initialize the CUDA context in each new process, which will fail.
RuntimeError: CUDA driver error: initialization error
I guess you are trying to re-initialize the CUDA context in each new process, which will fail.