Running out of memory Bandwidth

jonnedtc / U-Net-PyTorch

U-Net implementation in PyTorch for the LiTS challenge

22 stars 9 forks source link

Running out of memory Bandwidth #1

Open alizeshana opened 6 years ago

alizeshana commented 6 years ago

Is there a workaround to train the data on a 2GB GPU?

aarativasane89 commented 5 years ago

MemoryError Traceback (most recent call last)

in () 20 data= np.transpose(data, (2, 0, 1)) # transpose so the z-axis (slices) are the first dimension 21 #print(type(data)) ---> 22 output = np.zeros((len(data), 512, 512))[0] # save output here 23 #print(data) 24 # print(type(output)) MemoryError: >>how to solve these issue??