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

Question

我正在尝试在具有GPU的Windows上运行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

• nvcc 9.2 V9.2.148

• Windows 10版本1803内部版本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。