之前一直没有这种问题,重装了anaconda以后出现了这种问题。

在网上查了可能的原因,①CUDA版本与驱动程序不兼容;②CUDA库的路径设置存在问题;③PyTorch版本与CUDA版本不匹配;④编译问题;⑤软件包或库冲突。

解决方法在这篇博文中:【一文解决】已安装CUDA与Pytorch但torch.cuda.is_available()为False_torch.cuda.is available返回false-CSDN博客

不过这些和我的情况都不太符合。

我的解决办法是,重新安装了torch,解决了这个问题。显然没有缺失的包和版本冲突,所以故障原因不明。希望得到大佬的解答~

pip insatll D:\whl\torch\torch-1.12.0+cu113-cp39-cp39-win_amd64.whl

 

检查cuda是否可用的代码:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

Logo

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

更多推荐