这个错误一般是GPU的数目设置出现问题,我的代码如下所示:
THEANO_FLAGS=device=gpu3,floatX=float32,lib.cnmem=0.8
错误如下:
因为我的机器只有一个显卡,而我程序中写的是3,因此出现了这个错误,可以更改为cuda0如下:
这个错误一般是GPU的数目设置出现问题,我的代码如下所示:
THEANO_FLAGS=device=gpu3,floatX=float32,lib.cnmem=0.8
错误如下:
因为我的机器只有一个显卡,而我程序中写的是3,因此出现了这个错误,可以更改为cuda0如下: