使用 tensorflow-gpu 1.7.0 的问题

Question

我想测试一个建议的模型here，因为我需要 tensorflow-gpu 版本 1.7.0，我使用 pip 命令安装了所需的版本pip install tensorflow-gpu==1.7.0 一切运行，但是当我运行训练时出现以下错误

====NCBI_MODEL Training=====
Traceback (most recent call last):
  File "run.py", line 226, in <module>
    sess, batch_idx, infoInput=intOuts, tbWriter=tbWriter)
  File "/home/youssef/Documents/collabonet/CollaboNet-master/model/RunModel.py", line 82, in train1epoch
    feed_dict=feed_dict1)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 905, in run
    run_metadata_ptr)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1140, in _run
    feed_dict_tensor, options, run_metadata)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1321, in _do_run
    run_metadata)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1340, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[2,15] = 1651 is not in [0, 2)
     [[Node: BC5CDR-diseaseWE/embedding_lookup = Gather[Tindices=DT_INT32, Tparams=DT_FLOAT, _class=["loc:@BC5CDR-diseaseWE/embed"], validate_indices=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](BC5CDR-diseaseWE/embed, _arg_X_0_0)]]

Caused by op u'BC5CDR-diseaseWE/embedding_lookup', defined at:
  File "run.py", line 119, in <module>
    modelDict[dataSet]['WE']=modelClass.we(args=modelDict[dataSet]['args'])
  File "/home/youssef/Documents/collabonet/CollaboNet-master/model/CollaboNet.py", line 104, in we
    reuse=False)
  File "/home/youssef/Documents/collabonet/CollaboNet-master/ops/embeddingOps.py", line 76, in embedding_lookup
    inputs_embed = tf.nn.embedding_lookup(embedding_table, inputs)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/ops/embedding_ops.py", line 327, in embedding_lookup
    transform_fn=None)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/ops/embedding_ops.py", line 151, in _embedding_lookup_and_transform
    result = _clip(_gather(params[0], ids, name=name), ids, max_norm)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/ops/embedding_ops.py", line 55, in _gather
    return array_ops.gather(params, ids, name=name)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/ops/array_ops.py", line 2698, in gather
    params, indices, validate_indices=validate_indices, name=name)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/ops/gen_array_ops.py", line 2672, in gather
    validate_indices=validate_indices, name=name)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
    op_def=op_def)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 3290, in create_op
    op_def=op_def)
  File "/home/youssef/.local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1654, in __init__
    self._traceback = self._graph._extract_stack()  # pylint: disable=protected-access

InvalidArgumentError (see above for traceback): indices[2,15] = 1651 is not in [0, 2)
     [[Node: BC5CDR-diseaseWE/embedding_lookup = Gather[Tindices=DT_INT32, Tparams=DT_FLOAT, _class=["loc:@BC5CDR-diseaseWE/embed"], validate_indices=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](BC5CDR-diseaseWE/embed, _arg_X_0_0)]]

我怀疑它与tensorflow版本有关，所以当我运行import tensorflow as tf tf.__version__ 输出为'1.7.0' 我得出的结论是我得到的版本是 cpu 版本，但是我试图卸载它并重新安装它，同样的问题仍然存在。 我也尝试过 tensorflow-gpu 1.7.0 docker 映像，但它提供了相同的输出。 那么有什么办法可以克服这个问题吗？

Answer 1

问题似乎是由一个空文件引起的，结果导致一个空的提要字典。