默认安装，速度比较慢

pip install torch torchvision torchaudio  --extra-index-url https://download.pytorch.org/whl/cu118

百度了一下，有推荐这么改的，速度还是慢，因为还是转到官网 pytorch.org

pip install torch torchvision torchaudio --index-url https://mirror.sjtu.edu.cn/pytorch-wheels/cu118

最终采用 其中的版本 2.3.1，可以看看 pip 源

pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 torchaudio==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html