1. 首页
  2. 编程语言
  3. Python
  4. 解决TensorFlow训练内存不断增长,进程被杀死问题

解决TensorFlow训练内存不断增长,进程被杀死问题

上传者: 2022-03-27 05:18:07上传 PDF文件 52.13 KB 热度 13次

TensorFlow训练时,遇到内存不断增长,最终导致内存不足,进程被杀死。在TensorFlow中构造图,是将一些op作为节点加入图中,在run之前,是需要构造好一个图的,所以在run的时候,如果run图中不存在的节点,TensorFlow就会将节点加入图中,随着不断的迭代,造成内存不断的增长,从而导致内存不足。在上面的例子中,x,y是图中的两个节点,在while循环中run或者run都是不存在问题的,但是上例中我们使用的时run(x+y),而在TensorFlow的图中是不存在x+y这个节点的,所以在run的时候,图中会不断的创建该节点,最终导致了内存不足。

下载地址
用户评论