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。卸载并重新安装后，它可以再次工作。

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

首先下载用于CUDA 9.0的 CUDA 9.0 和 cuDNN v7，安装CUDA 9.0。

将cuDNN文件提取到您的CUDA安装文件夹中。（默认为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0）

将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin和C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64添加到PATH。

在tfjs-examples \ mnist-node中运行node main，现在可以使用。

第一次，Adding visible gpu devices: 0可能会停留2到5分钟，请耐心等待。我会的。

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

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

1 个答案: