torch检测不到gpu，无法使用cuda
一开始以为是cuda的版本不对，就开始不断的实验各种版本，后来发现可能是使用清华源下载时自动下载的cpu版本。查看torch版本的python命令，用pytorch官网自带的源下载。，发现输出的确为cpu版本。torch无法使用cuda。
·
torch无法使用cuda
报错检查：
一开始以为是cuda的版本不对，就开始不断的实验各种版本，后来发现可能是使用清华源下载时自动下载的cpu版本。
检查验证：
查看torch版本的python命令，import torch print(torch.__version__)，发现输出的确为cpu版本。
处理方法：
用pytorch官网自带的源下载。
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
8
0- 0
已为社区贡献1条内容
所有评论(0)