RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'

data
2

Hi,
Can you please see if this works -

torch.utils.data.DataLoader(
    ...,
    generator=torch.Generator(device='cuda'),
)
3 Likes