admin管理员组文章数量:1794759
关于电脑有独立显卡但torch.cuda.is
问题描述:
在安装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安装版本兼容。
版权声明:本文标题:关于电脑有独立显卡但torch.cuda.is 内容由林淑君副主任自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.xiehuijuan.com/baike/1686992747a126379.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论