tfjs-node-gpu неверный заголовок ELF в Windows

Question

Я пытаюсь запустить tenorflow / tfjs examples / mnist-node в Windows с графическим процессором.

После клонирования репозитория и установки зависимостей я могу запустить пример с процессором. Однако, когда я попытался установить @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 и cuDNN v7 для CUDA 9.0 , установить 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. запустите node main в tfjs-examples \ mnist-node, теперь это будет работать.

В первый раз это может занять от 2 до 5 минут, застряв на Adding visible gpu devices: 0, просто подождите. Я буду работать.

Кстати, вы еще не можете использовать графический процессор в режиме WSL.