国内安装pytorch gpu最简单方法

2024-10-07  本文已影响0人  AntiGravity

cmd直接安装容易断线,且无法续传,用梯子也不稳定,conda和pip效果差不多。
用国内源经常没有最新版,且径自下载cpu版本,体验感糟糕。
因此最简单的方法是去pytorch官网,找到pip下载的指定链接,直接去该网址下载whl文件,后本地直接安装。
以最新的cu124为例:
pip的--index-url为:https://download.pytorch.org/whl/cu124
直接进入,分别下载torch torchvision torchaudio等相应版本。

上一篇 下一篇

猜你喜欢

热点阅读