大佬可以跳过不看了,这里只是我试了后记录一下的。
原因:在下载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版本。
第七步,它可能会提示你,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/
转载文章受原作者版权保护。转载请注明原作者出处!