用pip安装cuda版本的torch

大佬可以跳过不看了,这里只是我试了后记录一下的。

原因:在下载torch_parse这个包的时候,它不知道怎么抽风了,把我的GPU版本顶掉了。我的猜测是可能因为文件名的冲突,它觉得我没有torch,就自动给我下了一个,结果莫名其妙地给我顶掉了。

途中花了很长时间来弄,我自身conda安装一直出错,镜像源不知道为什么之前出了问题还是没法解决,遂决定使用pip安装。

第一步:需要确定自己的cuda版本,可参考下面这个链接Win10安装CUDA11.4和Torch1.9.0

第二步:我的cuda版本是11.4,然后去pytorch官网上找对应版本,pytorch官网

 点击这个蓝色框的内容,然后跳转到如下界面

找到你对应的CUDA版本号对应的torch版本,以及torchvision

第三步:打开下面这个连接 :https://download.pytorch.org/whl/torch/

打开后是这样的

 

比如我是CUDA11.4,那么为了保险起见,我选择下载CUDA11.3版本对应的torch,也就是pytorch1.11.0+cu113。这些话是什么意思呢?

以橙色画到的这行为例,就是下载1.11.0的cuda版本的torch,python3.8,windows版本。

第四步:点击完成下载后,你会得到这样的一个文件

第五步:进行解压,按下win+R,输入cmd,打开终端,并激活环境 

 第六步,进入.whl文件所在的文件夹,

第七步,它可能会提示你,torchvision,torchaudio版本更不上之类的,你可以这么做:

pip install torchvision== -i https://pypi.tuna.tsinghua.edu.cn/simple/

 这样就可以啦

 

物联沃分享整理
物联沃-IOTWORD物联网 » 用pip安装cuda版本的torch

发表评论