The conv layers in PyTorch expect the input to have the channels in dim1 (channel_first).
Based on the error message it looks like you are passing the input as channel_last.
You could permute it using:
images_data_tensor = images_data_tensor.permute(0, 3, 1, 2).contiguous()