You could use a seed for the random number generator (torch.manual_seed) and make sure the split is the same every time.
Alternatively, you could split the sample indices, store each index tensor locally via torch.save, and use it in Subset.