pytorch-wheels方式安装:
pytorch-wheels 是 PyTorch pip 源的镜像。直接将 PyTorch 安装指引 中的 https://download.pytorch.org/whl 替换为 https://mirror.sjtu.edu.cn/pytorch-wheels 即可。
pip install torch==2.3.1+cu121 -f https://mirror.sjtu.edu.cn/pytorch-wheels/torch_stable.html