我在面部表情识别中使用以下GPU代码: http://code.google.com/p/deep-learning-faces/
我的系统与作者相同'但CUDA版本比作者高出一点(5.5)。 (5.0)。
当我运行" script_face_exp.m"时,它说"使用mexcuConvNNoo Assertion失败错误:cudaSetDevice错误!'。 它的原因是: cudaErrorInvalidDevice这表示用户提供的设备序号与有效的CUDA设备不对应。
但我可以使用GPU进行计算,基于theano的代码可以很好地运行在GPU上。
如何解决使用GPU正确运行代码的问题?
答案 0 :(得分:4)
script_face_exp
调用函数gg
来设置这些全局变量:
function [] = gg( gpunumber )
global GPUID;
global FERMI;
GPUID = 3;
FERMI = 1;
稍后在GPUID
中使用全局变量SelectGPUFromServer
来设置使用的GPU。
您可能没有ID为3的GPU(这意味着至少安装了4个GPU),因此您应该更改gg
以将此变量设置为例如0
(如果你有一个GPU)。