Solved by ptrblck in post #2 
              
              
                You have to assign the images and labels when you push them onto the GPU: 
images = images.to(device)
labels = labels.to(device)

Could you change it and try it again?