Solved by ptrblck in post #4 
              
              
                It should work with any size as long as your system can handle it properly. 
If you are using np.float32 values, the data should take approx 2GB of RAM. 
Are you using multiple workers in your DataLoaders? 
I’m not sure, what limitations Colab has on the RAM. 
Also, could you try to use torch.from_n…