We should add latest pytorch/torchvision installation from github into requirements, then eliminate all copy-pasted code.