前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >关于电脑有独立显卡但torch.cuda.is_available()运行出现为False的问题解决方法

关于电脑有独立显卡但torch.cuda.is_available()运行出现为False的问题解决方法

作者头像
算法与编程之美
发布2022-10-28 13:44:04
5K0
发布2022-10-28 13:44:04
举报

问题描述:

在安装conda环境后,确定自己电脑有独立显卡mx350,通过命令conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge尝试安装pytorch。但是在运行命令print('GPU存在:',torch.cuda.is_available()),输出一直为False,说明未能检查到电脑显卡。

解决方法:

1.首先想到的是会不会是安装pytorch出现了问题,准备将pytorch卸载了重装。但是在重新安装后问题依然不能够解决。

2.在网络上查阅相关文档后,猜测可能是自己CUDA版本不兼容的问题。于是更新了显卡驱动,将CUDA版本从10.2更新到了11.7。再次通过运行命令print('GPU存在:',torch.cuda.is_available()),输出为True,说明能检查到电脑显卡,问题解决。

总结:

针对安装pytorch不能够正常读取显卡的问题,首先不应该考虑的是安装pytorch出问题,如果安装pytorch真的发生了错误,相关的代码肯定是不能够运行的,但是在实际代码中又能够输出结果。

要通过关键词去搜索相关文档,才能够快速的发现问题所在。对显卡的驱动进行更新,让CUDA版本与pytorch安装版本兼容。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2022-09-24,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 算法与编程之美 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 问题描述:
  • 解决方法:
  • 总结:
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档