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

【问题标题】：How to solve cudaSetDevice Error and run this deep-learning code correctly?如何解决 cudaSetDevice 错误并正确运行此深度学习代码？
【发布时间】：2015-09-24 18:43:02
【问题描述】：

我在面部表情识别上使用以下 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正确运行代码？

【问题讨论】：

你安装了哪款 GPU？

标签： matlab cuda deep-learning conv-neural-network

【解决方案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 以将此变量设置为例如0（如果你有一个 GPU）。

【讨论】：