相关疑难解决方法(0)

如何在TensorFlow中解决'用完gpu内存'问题

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

tensorflow

27
推荐指数
5
解决办法
4万
查看次数

标签 统计

tensorflow ×1