如何防止在GPU PyTorch 0.3上建立内存使用情况

Question

使用此代码，经过几次迭代，我遇到了内存不足的错误。我已经隔离了将问题追加到self.ground_truth变量，我认为由于某种原因它仍然存储在GPU中。 x和gt_class_ids在GPU上开始。我正在尝试将其加载到CPU中，以便self.ground_truth变量不会占用GPU上的内存。我是PyTorch和Python的新手，并且仅限使用0.3版

x = x.cpu()[0]
gt_class_ids = gt_class_ids.cpu()[0]

for feature in x:
    self.ground_truth.append(feature)
for id in gt_class_ids:
    self.ground_truth_ids.append(id)

torch.cuda.empty_cache()