pip3 install -U torch torchvision --index-url https://download.pytorch.org/whl/cpu