TensorFlow 限制显存使用量

对于高配的计算机,可能都不用考虑限制显存这个问题,但是我要考虑。我轻薄本就 2GB 独显,默认加载数据时会一次性全部往显存里写,结果动不动就提示显存不够,然后程序 game over。

另外就是对于开发板,我这里也在用 Jetson Nano 的开发板,GPU 没有独立显存,是和 CPU 共用[……]

阅读全文