我在TensorFlow中运行了MNIST演示,其中包含2个转换层和一个完整的连接层,我得到了一条消息"用尽了分配2.59GiB的内存",但它显示总内存为4.69GiB,可用内存为3.22 GiB,它怎么能用2.59GiB停止?而对于更大的网络,我该如何管理gpu内存?我只关心如何充分利用gpu内存并想知道它是如何发生的,而不是如何预先分配内存
tensorflow
tensorflow ×1