Your labels should also be tensors as otherwise you couldn’t even push them onto your GPU:
labels = labels.cuda()

Are your labels currently stored in a list or a numpy array?
For the former you could just wrap them using torch.tensor(label) in your Dataset, for the latter use torch.from_numpy(label).