如何正确解决 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)。
我在面部表情识别上使用了以下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)。