tfjs-node-gpu在Windows上无效的ELF标头

Question

我正试图在Windows上用GPU运行tensorflow/tfjs-examples/mnist-node。

在克隆存储库并安装依赖项之后，我可以使用CPU运行该示例。但是，当我尝试安装@tensorflow/tfjs-node-gpu并运行示例时，这是我得到的：

Registration of backend tensorflow failed
Error: The specified module could not be found.    
\\?\C:\Users\noob\Desktop\tfjs-examples\mnist-node\node_modules\@tensorflow\tfjs-node-gpu\build\Release\tfjs_binding.node
    at Object.Module._extensions..node (internal/modules/cjs/loader.js:718:18)
    at Module.load (internal/modules/cjs/loader.js:599:32)
    at tryModuleLoad (internal/modules/cjs/loader.js:538:12)
    at Function.Module._load (internal/modules/cjs/loader.js:530:3)
    at Module.require (internal/modules/cjs/loader.js:637:17)
    at require (internal/modules/cjs/helpers.js:20:18)
    at bindings (C:\Users\noob\Desktop\tfjs-examples\mnist-node\node_modules\bindings\bindings.js:81:44)
    at C:\Users\noob\Desktop\tfjs-examples\mnist-node\node_modules\@tensorflow\tfjs-node-gpu\dist\index.js:10:60
    at Environment.registerBackend (C:\Users\noob\Desktop\tfjs-examples\mnist-node\node_modules\@tensorflow\tfjs-core\dist\environment.js:188:27)
    at Object.<anonymous> (C:\Users\noob\Desktop\tfjs-examples\mnist-node\node_modules\@tensorflow\tfjs-node-gpu\dist\index.js:9:9)

根据tfjs-node，似乎我已经符合要求。这是我的环境：

环境：

• 所以v10.9.0
• 等人
• Windows 10版本1803 build 17738.1000

Answer 1

好的，问题解决了。我必须安装CUDA v9.0而不是v9.2。卸载并重新安装后，它再次工作。

如果有人有类似的问题，这里有一些教程。

1. 首先下载CUDA 9.0的CUDA 9.0和cuDNN v7，安装CUDA 9.0。
2. 将cuDNN文件解压缩到CUDA安装文件夹。 （默认为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0）
3. 将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64添加到PATH。
4. 在tfjs-examples \ mnist-node中运行node main，现在可以正常工作了。

第一次，可能需要大约2~5分钟卡在Adding visible gpu devices: 0，等待它。我会工作的

顺便说一下，你还不能在WSL模式下使用GPU。