【发布时间】:2018-05-31 07:00:20
【问题描述】:
所以我在一台装有GeForce GTX 980显卡的Windows 10机器上安装了GPU版本的TensorFlow。
诚然,我对显卡知之甚少,但根据 dxdiag 的说法,它确实有:
4060MB 的专用内存 (VRAM) 和;
8163MB 共享内存
总共约12224MB。
不过,我注意到的是,这种“共享”内存似乎毫无用处。当我开始训练模型时,VRAM 将填满,如果内存需求超过这些 4GB,TensorFlow 将崩溃并显示“资源耗尽”错误消息。
当然,我可以通过选择适当低的批量大小来防止达到这一点,但我想知道是否有办法利用这些“额外”8GB 的 RAM,或者是否就是这样而 TensorFlow 需要要专用的内存。
【问题讨论】:
标签: tensorflow shared-memory vram