如何将tfjs-node与从源代码构建的libtensorflow一起使用

时间:2018-10-19 06:47:21

标签: tensorflow node-gyp tensorflow.js

我看到可以使用从源代码构建的libtensorflow,如自述文件所述 this thread on the MSIX community

我已经成功地从源代码构建了版本,但是我不知道如何让tfjs-node使用此自定义版本,而不是预先构建的版本。

1 个答案:

答案 0 :(得分:2)

我发现以下解决方法似乎可行:

从tensorflow源目录复制文件bazel-bin/tensorflow/libtensorflow.so,以替换项目内的node_modules/@tensorflow/tfjs-node/deps/lib/libtensorflow.so

希望有一种更好的方法来npm install,这样就不必下载预编译的版本。