I tried it on both my laptop (screenshot from above) and on my Ubuntu machine (which has the cuda version of PyTorch). It seems to work in both cases. Regarding the val_loader, just enabled it:

Screen Shot 2019-04-18 at 12.30.36 PM.png708×687 73.5 KB