0

如题,我准备把3万张左右的图片先保存成numpy格式,这样以后再用的时候会方便很多,速度也快一些。

但是在运行的过程中貌似是因为显存不足报错了,是在读取到90多张图片左右就报错了,下面是我的代码,请问有什么问题吗?

(我用了实验室的服务器,显存有10G左右)

代码:


x = None
y = None
i = 0
for data in dataloaders[key]:
    # get the inputs
    inputs, labels = data
    
    print("Num {}".format(i))
    if i == 0:
        x = inputs.cuda()
        y = labels.cuda()
    else:
        x = torch.cat((x,inputs.cuda()))
        y = torch.cat((y,labels.cuda()))
    i += 1


np.save('./data/X_fixed.npy', x.cpu().numpy())
np.save('./data/Y_fixed.npy', y.cpu().numpy())

运行结果:

clipboard.png

marsggbo 226
2018-04-02 提问
0 个回答

撰写答案

推广链接