在同一Python进程中运行多个Tensorflow会话时出错 [英] Error while running multiple Tensorflow sessions in the same Python process

问题描述

我有一个具有以下层次结构的项目:

I have a project with this hierarchy:

project
├── libs
│   ├── __init__.py
│   ├── sub_lib1
│   │   ├── file1.py
│   │   └── __init__.py
│   └── sub_lib2
│       ├── file2.py
│       └── __init__.py
└── main.py

main.py的内容:

from libs.sub_lib1.file1 import func1
from libs.sub_lib2.file2 import func2


#some code

func1(parameters)

#some code

func2(parameters)

#some code

file1.py的内容:

#import some packages
import tensorflow as tf

def func1(parameters):

    #some code

    config = tf.ConfigProto()
    config.gpu_options.allow_growth=True
    tf.reset_default_graph()
    x = tf.placeholder(tf.float32,shape=[None,IMG_SIZE_ALEXNET,IMG_SIZE_ALEXNET,3])
    y_true = tf.placeholder(tf.float32,shape=[None,output_classes])
    with tf.Session(config=config) as session:
        saver.restore(session, "path to the model1")
        k = session.run([tf.nn.softmax(y_pred)], feed_dict={x:test_x , hold_prob1:1,hold_prob2:1})
    #some code
    return(the_results)

file2.py的内容:

#import some packages
import tensorflow as tf

def func2(parameters):

    #some code

    config = tf.ConfigProto()
    config.gpu_options.allow_growth=True
    sess = tf.Session(config=config)
    with gfile.GFile('path the model2', 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        sess.graph.as_default()
        tf.import_graph_def(graph_def, name='')
    sess.run(tf.global_variables_initializer())
    #Get the needed tensors 
    input_img = sess.graph.get_tensor_by_name('Placeholder:0')
    output_cls_prob = sess.graph.get_tensor_by_name('Reshape_2:0')
    output_box_pred = sess.graph.get_tensor_by_name('rpn_bbox_pred/Reshape_1:0')
    #some code to prepare and resize the image
    cls_prob, box_pred = sess.run([output_cls_prob, output_box_pred], feed_dict={input_img: blobs['data']})
    #some code
    return(the_results)

运行main.py时，出现以下错误:

When I run main.py, I got the following error:

Traceback (most recent call last):
  File "main.py", line 46, in <module>
    func2(parameters)
  File "/home/hani/opti/libs/sub_lib2/file2.py", line 76, in func2
    cls_prob, box_pred = sess.run([output_cls_prob, output_box_pred], feed_dict={input_img: blobs['data']})
  File "/home/hani/.virtualenvs/opti/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 929, in run
    run_metadata_ptr)
  File "/home/hani/.virtualenvs/opti/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1128, in _run
    str(subfeed_t.get_shape())))
ValueError: Cannot feed value of shape (1, 600, 863, 3) for Tensor 'Placeholder:0', which has shape '(?, 227, 227, 3)'

经过一些调试，我在第二个模型中没有找到任何形状为 (?, 227, 227, 3) 的张量.相反，我发现张量x(由 x = tf.placeholder(tf.float32，shape = [None，IMG_SIZE_ALEXNET，IMG_SIZE_ALEXNET，3]定义)在文件1的func1中具有(?，227，227，3)作为形状.我从文件file2中的func2中检查了input_img的形状(由 input_img = sess.graph.get_tensor_by_name('Placeholder:0')定义)，我发现它的形状(?，227，227，3)当我运行main.py.但是，当我运行file2.py(通过运行 python file2.py 独立地)时，我没有收到此错误，并且发现input_img的形状是占位符形状:(?，?，?，3).
所以我假设可能两个模型都具有相同的张量名称( placeholder )，并且当我在main.py中导入file1和file2时，占位符的第一种形状(?，227，227，3)保留在GPU内存中.
我在file1.py中尝试了 session.close()，但这没有用！
有没有更合适的方法在同一过程中使用多个Tensorflow会话而又不会引起它们之间的任何混乱呢?或者简单地说，如何在同一python进程中启动另一个Tensorflow会话之前正确关闭它?

After some debugging, I didn't find any tensor in the second model that has (?, 227, 227, 3) as shape. Instead, I found that the tensor x (defined by x = tf.placeholder(tf.float32,shape=[None,IMG_SIZE_ALEXNET,IMG_SIZE_ALEXNET,3]) in func1 from file1) has (?, 227, 227, 3) as shape. I checked the shape of input_img (defined by input_img = sess.graph.get_tensor_by_name('Placeholder:0') in func2 from file file2), I found it (?, 227, 227, 3) when I run main.py. However when I run file2.py (independently by running python file2.py), I don't get this error and and I found that the input_img's shape is placeholder shape: (?, ?, ?, 3).
So I assumed that may be both models have the same tensor's name (placeholder) and when I import both file1 and file2 in main.py, the first shape of placeholder (?, 227, 227, 3) remains in the GPU memory.
I tried session.close() in file1.py, but it didn't work!
Is there a more appropriate way to use multiple Tensorflow sessions in the same process without getting into any confusion between them? Or simply, how to properly close a Tensorflow session before starting another one in the same python process?

推荐答案

阅读了Stack Overflow中的一些相关文章之后，我在我引用的答案:

After reading some related posts in Stack Overflow, I found a solution in this answer from which I quote:

由于尝试执行以下操作，您可能在第二个build_graph()期间出错创建具有相同名称的变量(在您的情况下会发生这种情况)，图正在定稿等

you might get errors during second build_graph() due to trying to create variables with the same names (what happens in your case), graph being finalised etc.

要解决我的问题，我只需要在main.py中添加 tf.reset_default_graph()即可重置图形及其参数.

To solve my problem, I only needed to add tf.reset_default_graph() to main.py in order to reset the graph and its parameters.

这篇关于在同一Python进程中运行多个Tensorflow会话时出错的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！