在 GPU 上运行火炬预测模型时出错答案

【问题标题】：Error when running torch prediction model on GPU在 GPU 上运行火炬预测模型时出错
【发布时间】：2018-06-30 04:54:22
【问题描述】：

我一直在尝试使用特定的预训练机器学习模型来为图片添加字幕。我一直在使用https://github.com/unnonouno/densecap。

它带有一个 Dockerfile，用于设置整个 cuda/torch/cudnn 环境。

通过运行run_model.lua 脚本对新图片进行预测。它在通过传递-gpu -1 在 CPU 上运行时确实有效，但在消除争论并在 GPU 上运行时无效。在这种情况下，我收到以下错误：

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

我尝试了不同的方法，例如通过运行luarocks install cudnn 重新安装cudnn 或从cudnn5 降级到cudnn4 均未成功。

【问题讨论】：

标签： lua dockerfile cudnn

【解决方案1】：

问题似乎与您的 CUDA 驱动程序有关：

CUDA 驱动程序版本不足以满足 CUDA 运行时版本

看看类似的讨论here。

无需更改您的 cuDNN 版本。您只需要纠正您的 CUDA 驱动程序/工具包兼容性。

【讨论】：