Проблема со сборкой caffe с CUDA + cuDNN
Всем привет, собрал caffe по следующим шагам. В начале решил собрать с CPU-ONLY=1. Немного поработав со сборкой решил пересобрать с поддержкой вычислений на GPU, т.к. мой intel core i5 9300h грузился почти до 100%. Потом:
- Установил графический драйвер (NVIDIA GeForce GTX 1650 Max-Q)
- Установил CUDA 9.0
- Скачал 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 сборки)