Looking at the DataLoader sources, it seems that CPU->GPU data transfers are triggered within a thread instead of a process. Are there any restrictions against calling Tensor.to(“cuda:0”) or Tensor.copy_() from within a worker process? What about Tensor.pin_memory()?