解决Python中Torch安装问题的完美攻略

大家一直都会遇到python 的torch安装问题。答案很多,试了很久,可就是装不好。原因是版本会更新,大家的环境各不相同,遇到的问题也各不相同。那有没有一种通用的办法可以彻底解决呢?当然是有的!今天我用一篇攻略教你如何解决torch的安装,找到最核心的思路,到底是哪里的问题导致你就是装不好。
0.卸载当前的torch版本。怎么查看呢?
import torch
print(torch.version)
1.不要相信自动安装,包括pytorch官网的!
pytroch地址:https://pytorch.org/

这个地址你可以自行选择自己的系统或者环境。安装方式等。点击后。Run this Command:里会生成你要用的安装指令。
2.那如何选择你的cuda版本呢?
win+R 输入cmd 按回车,再输入nvidia-smi按回车查看

3.这样你从Run this Command里拿到的命令肯定能安装。但是坑来了。终端执行的时候,一定要看好,它跳转到了什么下载地址。如果你是38的python。他要给你下载310的torch,显然是不合适的。你可以选择装个310的环境或者,或者直接关掉终端,用whl方式自己安装

地址:https://download.pytorch.org/whl/nightly/torch/,如果你要装win的,就ctrl+F搜索win。找到最后一个!再看你的python版本,比如3.9python就选箭头所指。点击下载。终端执行pip install +绝对路径,

这样装出来的torch,不会出任何问题。也不会是+cpu版本

物联沃分享整理
物联沃-IOTWORD物联网 » 解决Python中Torch安装问题的完美攻略

发表评论