Tensorflow.js looks for CUDA 10.0 with libtensorflow recompiled against CUDA 10.1

To get help from the community, we encourage using Stack Overflow and the tensorflow.js tag.

TensorFlow.js version

2.0.0

Browser version

n/a, Node.js v14.2.0

Describe the problem or feature request

If I recompile libtensorflow against CUDA 10.1.168 according to the instructions here and substitute it in as instructed into my Tensorflow.js install, it is observed that Tensorflow.js continues to fail to load the CUDA 10.0 binaries.

For the exact steps I followed to compile libtensorflow, see this comment I made on a different issue. Following that, I did the following:

cp bazel-bin/tensorflow/tools/lib_package/libtensorflow.tar.gz ~/myproject/node_modules/@tensorflow/tfjs-node/deps
cp bazel-bin/tensorflow/tools/lib_package/libtensorflow.tar.gz ~/myproject/node_modules/@tensorflow/tfjs-node-gpu/deps
cd ~/path-to-my-project/node_modules/@tensorflow/tfjs-node/deps
tar -xf libtensorflow.tar.gz
cd ~/path-to-my-project/node_modules/@tensorflow/tfjs-node-gpu/deps
tar -xf libtensorflow.tar.gz

…as instructed by your README (link above).

Specifically, I’m getting this error: https://ybin.me/p/2007f2550ecf75a4#m91xA9jifKJumfGEs1YOpANyeuRKrtTI4m7SnfgdM6I=

(it’s a lot of output, so I used a pastebin).

To be clear, I recompiled libtensorflow against CUDA 10.1, but even after substituting it in Tensorflow.js complains it can’t find CUDA 10.0.

Code to reproduce the bug / link to feature request

Any example Tensorflow.js program for Node.js (it doesn’t even have to do anything).

[spazmodius]

I have found (I’m on Windows) that I have to replace the binary in the lib/napi folder, not deps.

[google-ml-butler[bot]]

Are you satisfied with the resolution of your issue? Yes No