用pip安装cuda版本的torch

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

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

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

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

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

用pip安装cuda版本的torch

用pip安装cuda版本的torch

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

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

用pip安装cuda版本的torch

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

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

用pip安装cuda版本的torch

用pip安装cuda版本的torch

用pip安装cuda版本的torch

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

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

这样就可以啦

Original: https://blog.csdn.net/wwee1223/article/details/126726014
Author: wwee1223
Title: 用pip安装cuda版本的torch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/671795/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球