你可以在 TensorFlow 配置期间选择建议的默认值。注意:这不会修改你现有的 TensorFlow 安装。这一步是让 Sonnet 构建于 TensorFlow 头文件上的必要步骤。 构建和运行安装程序 运行安装脚本,在临时目录中创建一个 wheel file: $ mkdir /tmp/sonnet $ bazel build --config=opt :install $ ./bazel-bin/install /tmp/sonnet pip install 生成的 wheel file: $ pip install /tmp/sonnet/*.whl 如果已安装 Sonnet,卸载 wheel file 上之前的 pip install : $ pip uninstall sonnet 你可以通过尝试重采样(resampler op)这样的操作来验证 Sonnet 是否安装成功: $ cd ~/ $ python>>> import sonnet as snt>>> import tensorflow as tf>>> snt.resampler(tf.constant([0.]), tf.constant([0.])) 预计的输出应该是: <tf.Tensor 'resampler/Resampler:0' shape=(1,) dtype=float32> 当然,如果引入了 ImportError,C++组件未找到时,请确保你没有导入复制的源代码(即在复制的资源库外调用 Python),并在安装 wheel file 前卸载 Sonnet。 原文链接:https://deepmind.com/blog/open-sourcing-sonnet/ 本文为机器之心编译,转载请联系本公众号获得授权。 ?------------------------------------------------ 加入机器之心(全职记者/实习生):[email protected] 投稿或寻求报道:[email protected] 广告&商务合作:[email protected] (责任编辑:本港台直播) |