问题描述:NVIDIA -SMI的版本是12.5,cuda的版本是11.8,path路径都没有问题

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include

通过nvcc --version查询版本
在这里插入图片描述
可以发现cuda的配置是没有问题的
但是python里面

 import torch
 torch.cuda.is_available()

显示是false
最后发现问题出在了pytorch版本上,可以通过conda list查看一下
在这里插入图片描述
原因是我版本后面这里是cpu的(此处照片是改正后的)
大家可以去pytorch官网https://pytorch.org/
在这里插入图片描述用conda部分的代码重新安装,我这里因为本身带着cuda,然后用pip下载cuda11.8发现下载的只有100mb左右的更新,然后并不能解决问题,用conda的时候,他将我的pytorch从2.3.0更新到2.3.1,下载1g的文件之后解决了。

总的来说,如果你的cuda配置好了,然后和pytorch版本也是对应的,但是还是搜不到gpu,就请查看下list里面pytorch是不是下载成了cpu版本。然后予以更正就好了。有时候下载不了可能是channel的问题,可以添加下清华源等这些,然后再更新pytorch。

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐