1.问题
很久没用pytorch,今天在使用以下命令查看下GPU是否可以获取,显示为false,检测后发现显卡数量为0.
import torch
torch.cuda.is_available()
torch.cuda.device_count()
但是我之前已经通过anaconda配置好虚拟环境且成功运行过多次,不考虑显卡本身坏掉的问题,最后查询到是显卡被禁用了。
2.解决方案
win+r打开运行,输入devmgmt.msc打开设备管理器(或者在win11搜索框输入设备管理器直接搜索)
向下拉直到展开显示适配器,右键单击第二项
点击启用设备即可打开GPU
此时在命令行输入nvidia-smi即可看到具体信息,若不能看到具体信息则重启后再次尝试即可。