Upgrade to latest stable pytorch and torchvision

[daniel-j-h]

• [ ] Bump packages for both cpu as well as gpu docker images.
• [ ] Re-calculate dependencies and lock dependency tree via pip-compile.
• [ ] Test end to end

Don't expect anything breaking for this version update; but we need to test end to end.