Make UICollectionViewCells with UIImageView height based on image

Question

I want to make a UICollectionView with the cells the width of the screen, but the height is based on the aspect ratio of the UIImageView inside.