torch.cuda.device_count()只检测到一个gpu

Question

今天用gpu加速训练网络模型 但是我打开资源管理器看到每次都是集成显卡在运行 我的RTX2060利用率是0 设置device('cuda:1')会报错 我就打印了一下看看我有几个能用的显卡

print(torch.cuda.device_count())

print(torch.cuda.is_available())

结果

1
True

发现只有一个能用的 这是为什么呀 有没有人指点一下

Answer 1

你的集成显卡只是用于桌面应用的展示，也没有用于训练，集成显卡的性能是很差的。torch.cuda.device_count()只会找独立显卡，是找不到系统的集成显卡的，你现在的代码完全是使用cpu在训练，如果要使用gpu做训练。则需要将数据和模型都迁移到cuda上。用法：data=data.cuda(), model=model.cuda()