pip install --upgrade torch==1.4.0 torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

包名自己修改即可。