【问题标题】:Running SyntaxNet with designated instance (in Python-level)使用指定实例运行 SyntaxNet(在 Python 级别)
【发布时间】:2016-11-15 06:18:34
【问题描述】:

您能否告诉我在训练/测试 SyntaxNet 时如何指定要使用的实例?

在其他 tensorflow 模型中,我们可以通过编辑 Python 代码轻松更改配置:

例如)tf.device('/cpu:0') => tf.device('/gpu:0')

我可以通过运行 demo.sh 来运行 parsey mcparseface 模型,然后我返回符号链接来查找设备配置。

也许我错了但是我在demo.shparser_eval.pycontext.proto中找不到gpu配置python代码。

当我在 tensorflow/models 中使用查询“device”进行搜索时,我可以看到几个 C 文件,例如 syntaxnet/syntaxnet/unpack_sparse_features.cc 包含行 using tensorflow::DEVICE_CPU;

那么.. 更改这些文件中的 C 代码是更改 SyntaxNet 设备配置的唯一方法吗?

我希望有更简单的方法来更改 Python 级别的设置。

提前致谢。

【问题讨论】:

    标签: python nlp tensorflow gpu syntaxnet


    【解决方案1】:

    在 GPU 上运行语法网络的说明可以参考这个页面:https://github.com/tensorflow/models/issues/248

    Tensorflow 会自动将包括 GPU 在内的设备分配给操作:https://www.tensorflow.org/versions/r0.11/how_tos/using_gpu/index.html。您也可以在构建图表时手动指定设备。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-01-09
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多