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。