问题
在安装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安装版本兼容。