Sorry, my mistake  I forgot to add this line to use GPU device = torch.device("cuda:0")



 ptrblck:

A single image in torch.uint8 would take approx 5100*3500*3/1024**2 = 51MB .


This might help me with another project idea! Pytorch doesn’t support .CR2, RAW image, file formats yet, right?