Мне удалось успешно скомпилировать Caffe на плате Nvidia Jetson TX1 с CUDA 9.0 и Open CV 3.
Однако, когда я запускаю следующую команду для тестирования Caffe:
build/tools/caffe time --model=models/bvlc_alexnet/deploy.prototxt --gpu=0
Я получаю следующую ошибку:
F0712 23:05:53.664676 28580 im2col.cu:61] Check failed: error == cudaSuccess (48 vs. 0) no kernel image is available for execution on the device
Если я уберу флаг --gpu=0
, я больше не увижу ошибки.
Любая помощь / предложения о том, как я могу получить код для использования графического процессора, будут очень признательны.