训练模型的时候显示GPU可使用内存不够，打开任务管理器查看内存是够用是什么情况？

如图，显示训练模型需要的GPU显存不够，大部分被占用，但是我打开任务管理器发现根本没有被占用，网上查的减小batch_size，但是我减到1了还是报同样的错误，连需要的显存都一样，这是什么原因要怎么修改？

我的猜测，你可能有一块集成显卡，但没有禁用

程序跑起来一直看着，直到报错，看看显卡内存是不是爆了？

请问你解决了吗？

需要你设置按需分配内存。可以查下一大堆代码。