Question about DataLoader and pin_memory_device

data
1

When looking at the source code on github, there is an option to specify on which device pin_memory is used. I need to specify which gpu my dataloader runs but when I try to add the pin_memory_device argument, I get

TypeError: DataLoader.init() got an unexpected keyword argument ‘pin_memory_device’

Any idea why I get this error?
Thanks!

2

The parameter was added to the master branch on Apr 20, 2022 and will be part of release 1.12 (which is coming out in ~2 weeks). If you would like to use that parameter, please install the nightly version of PyTorch.

3

Now I see why I get this error, thanks !