缩小不再需要的 GPU 内存对象

缩小不再需要的 GPU 内存对象

经过一番搜索,我没有找到任何标准的方法来释放分配给 Tensorflow 的 GPU 内存。所以我想知道有没有办法手动缩小不再需要的 VRAM 对象而无需实际重新启动?(它几乎占用了我所有的 VRAM)
操作系统的职责是在进程终止后释放系统内存,但它不能释放 GPU 内存,因为它已由 DMA 和图形驱动程序控制。

操作系统:Ubuntu 18.04
驱动程序:nVidia 430
GPU:GeForce GT 740M

任何帮助都将受到赞赏。

谢谢。

相关内容