1.下载网页:PyTorch

下拉到下面界面,cuda的版本查找可以参考CUDA学习(一)——如何查看自己CUDA版本?_cuda version-CSDN博客

我的版本号是12.7

然后把上面run this command输入到任务管理器:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

pip后的3可删可不删,崩溃装这么久。休息一天了,明天继续(慢的原因主要是在官网下如果有类似于镜像网站的话可能会快很多)

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

这种可能会更快,但我还是用原装了防止有问题。

晚上12点终于好了

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐