matlab - 如何解决 cudaSetDevice 错误并正确运行此深度学习代码？

Question

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

我的系统与作者的系统相同，但 CUDA 版本（5.5）比作者的（5.0）略高。

当我运行“script_face_exp.m”时，它说“Error using mexcuConvNNoo Assertion Failed: cudaSetDevice Error!”。这是由：cudaErrorInvalidDevice 引起的，这表明用户提供的设备序号与有效的 CUDA 设备不对应。

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

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

Answer 1

script_face_exp调用gg设置这些全局变量的函数：

function [] = gg( gpunumber )
global GPUID;
global FERMI;
GPUID = 3;
FERMI = 1;

全局变量GPUID稍后用于SelectGPUFromServer设置使用的 GPU。

您可能没有 ID 为 3 的 GPU（这意味着至少安装了 4 个 GPU），因此您应该更改gg将此变量设置为 eg 0（如果您有一个 GPU）。