誘因

• 最近 conda 太不給力了，清華源的 conda 總是給默認裝 cpu 版本，conda 自己速度又不太給力，所以干脆直接用 pip 解決
• 但是 pip 通常無法提供像 conda 一樣便捷的服務，需要考慮和解決很多版本適配的問題，所以這里介紹一種方法，直接 pip 一步到位

方法

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

詳情可以看這篇文章，但是這篇文章中給的命令稍微有點問題；用我這個就行~
https://blog.csdn.net/thequitesunshine007/article/details/118668749

總結

以上是生活随笔為你收集整理的环境配置之：pip 一步到位解决 pytorch-gpu 安装的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。