安装cpu版torch：pip3 install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
安装gpu版torch：pip3 install torch==1.2.0cu92 torchvision==0.4.0cu92 -f https://download.pytorch.org/whl/torch_stable.html（cu92为对应的cuda版本）
查看gpu版本是否安装成功：import torch
print(torch.cuda.is_available())
true即成功
查看cuda版本：cmd里输入nvcc -V