Update pytorch

relaxed constraints on both pytorch and torchvision:

• pytorch >= 0.4.0 (instead of ==)
• torchvision >= 0.2.0 (instead of ==)

Updated the CI (dropping out Python 2)