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

Posted 2023-01-20 算法与编程之美

问题

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