In the post, it is mentioned that pytorch creates a context that should be close to 250MB irrespective of the size of the variables I create. In my case, it seems like the same variable is being created on both the GPUs. If you look at the output of nvidia-smi, the process 7932 takes up 549MB on both GPUs. Is this the intended behavior?