如何正确解决 cudaSetDevice Error 和 运行 这个深度学习代码?

How to solve cudaSetDevice Error and run this deep-learning code correctly?

我在面部表情识别上使用了以下GPU代码: http://code.google.com/p/deep-learning-faces/

我的系统和作者的一样,只是CUDA版本(5.5)比作者的(5.0)高一点。

当我 运行 "script_face_exp.m" 时,它说“错误使用 mexcuConvNNoo 断言失败:cudaSetDevice 错误!”。 它是由以下原因引起的: cudaErrorInvalidDevice 这表示用户提供的设备序号不对应于有效的 CUDA 设备。

但我可以使用GPU进行计算,基于theano的代码在GPU上运行良好。

如何解决使用 GPU 来正确 运行 代码?

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)。