Python install torchvision download. DEVICE='cpu' in the config.