Pytorch 运行时错误：无法分配字节对象！在火炬上，救命

我现在在用torch保存张量数据时遇到了一个问题。 我的队友给了我代码，我试着用Jupyter笔记本运行它

sample_data = trainDataset(dir_data = train_path, mode_test = False)
X = list(map(lambda x: x[0], sample_data))
y = list(map(lambda x: x[1], sample_data))

b, c, w, h = X[0].shape

X_tensor = torch.Tensor(len(X), c, w, h)
torch.cat(X, out=X_tensor)
X_tensor.float()
y_tensor = torch.FloatTensor(y)

print("save Y")
torch.save(y_tensor, y_train_path)
print("save X")
torch.save(X_tensor, x_train_path)

根据队友的说法，X的数据约为12GB。但在保存X数据之前，会发生运行时错误。有谁知道这个问题的解决方法吗？

请阅读。显示异常的完整堆栈跟踪