Проблема со сборкой caffe с CUDA + cuDNN

Всем привет, собрал caffe по следующим шагам. В начале решил собрать с CPU-ONLY=1. Немного поработав со сборкой решил пересобрать с поддержкой вычислений на GPU, т.к. мой intel core i5 9300h грузился почти до 100%. Потом:

  1. Установил графический драйвер (NVIDIA GeForce GTX 1650 Max-Q)
  2. Установил CUDA 9.0
  3. Скачал cuDNN v7.3.1.20 перекинул .lib .dll .h в папку с CUDA

После запуска build_win в выводе консоли вижу, что cmake не обнаружил куду:

CMake Warning at cmake/Dependencies.cmake:97 (message):
   -- CUDA is not detected by cmake.  Building without it...

... и далее сборка для CPU.

Я проверил на наличие системной переменой:

C:\Users\smolk>set cuda
CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
CUDA_PATH_V9_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

Потом пробовал сам поискать где выставить флаги, но без результата. Подскажите что делать?(Я имею малый опыт в конфиге cmake сборки)


Ответы (0 шт):