Hi.
I specify a second GPU for training my model via DEVICE = torch.device("cuda:1" if torch.cuda.is_available() is True else "cpu") but the memory of the first GPU, which is cuda:0 is allocated. Can you please let me know what am I missing?

image955×630 18 KB