I believe DataParallel assumes identical or very similar GPUs give the _check_balance warning here:
torch.nn.parallel.data_parallel — PyTorch 1.10.0 documentation 1
However, you may want to check that using the 1050 is beneficial from a load-balancing perspective to begin with as it may be bottlenecking training compared to the P2000s even when accounting for the difference in available memory.