安装pytorch-gpu

pytorch-wheels 是 PyTorch pip 源的镜像。直接将 PyTorch 安装指引 中的 https://download.pytorch.org/whl 替换为 https://mirror.sjtu.edu.cn/pytorch-wheels 即可。