Python Tensorflow-使用IM2Text在GeForce 970上进行内存管理

我只是想知道是否有办法避免IM2Text使用GeForce 970 GTX的超低0.5GB。我们是否可以强制培训课程仅使用常规的3,5GB？这是个好主意吗

---

谢谢。

您可以尝试使用下面的命令限制内存总量。虽然不确定分配哪个内存，但如果您运气不好，它可能会包含0.5GB的慢速内存

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

---

我认为您无法选择要使用的特定GPU内存。看来970的内存分区有一些问题。您可以禁用GPU。为deeplearning严肃应用程序禁用GPU是没有意义的。为严肃应用程序使用GeForce 970是没有意义的。是一张（小车）游戏卡。拿回你的钱：不，不是。它仍然比任何CPU都要快，即使内存错误使它下降到580的水平。为了我的钱，我正在努力——没那么容易。谢谢，我一到家就去试试。@Mathieurbezhermoso你能分享一下结果吗？