Ошибка при запуске модели прогнозирования горелки на графическом процессоре - PullRequest
Новая
       32

Ошибка при запуске модели прогнозирования горелки на графическом процессоре

0 голосов
/ 28 июня 2018

Я пытался использовать специальную модель машинного обучения для подготовки субтитров к субтитрам. Я использовал https://github.com/unnonouno/densecap.

Он поставляется с Dockerfile, настраивающим всю среду cuda / torch / cudnn.

Прогнозы для нового изображения создаются с помощью сценария run_model.lua. Он работает при запуске его на процессоре, передавая -gpu -1, но не при удалении аргумента и запуске его на графическом процессоре. В этом случае я получаю следующую ошибку: 

THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-8398/cutorch/lib/THC/THCGeneral.c line=70 error=35 : CUDA driver version is insufficient for CUDA runtime version
/root/torch/install/bin/luajit: 
/root/torch/install/share/lua/5.1/trepl/init.lua:389: loop or previous error loading module 'cutorch'
stack traceback:
    [C]: in function 'error'
    /root/torch/install/share/lua/5.1/trepl/init.lua:389: in function 'require'
    ./densecap/utils.lua:26: in function 'setup_gpus'
    run_model.lua:145: in main chunk
    [C]: in function 'dofile'
    /root/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
    [C]: at 0x00406670

Я пробовал разные вещи, такие как переустановка cudnn с помощью runnign luarocks install cudnn или понижение с cudnn5 до cudnn4 без какого-либо успеха.

1 Ответ

0 голосов
/ 29 июня 2018

Похоже, проблема связана с вашим драйвером CUDA:

Версия драйвера CUDA недостаточна для версии CUDA runtime

Посмотрите на подобные обсуждения здесь .

Нет необходимости менять версию cuDNN. Вам просто нужно исправить совместимость драйвера и инструментария CUDA.

...