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

2024-01-01

我看到可以使用libtensorflow如自述文件中所述,是从源代码构建的https://github.com/tensorflow/tfjs-node#optional-build-libtensorflow-from-tensorflow-source https://github.com/tensorflow/tfjs-node#optional-build-libtensorflow-from-tensorflow-source

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


我发现以下解决方法似乎有效:

复制文件bazel-bin/tensorflow/libtensorflow.so从tensorflow源码目录进行替换node_modules/@tensorflow/tfjs-node/deps/lib/libtensorflow.so在你的项目里面。

希望有更好的方法来做npm install这样就无需下载预编译版本。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何将 tfjs-node 与从源代码构建的 libtensorflow 结合使用 的相关文章

随机推荐