使用theano,keras在Windows上使用多个gpus

时间:2017-03-11 22:31:03

标签: gpu theano keras theano-cuda

我是深度学习/ theano / keras的初学者。我试图弄清楚如何在Windows 7上使用多个gpus。我已成功安装Theano,keras(如本文所述) How do I install Keras and Theano in Anaconda Python on Windows?)并使用一个gpu。我想同时使用我的gpus

以下是配置和版本的详细信息

Python - 2.7(Anaconda-4.3.14,Windows-64bit) ,CUDA - 7.5.17 ,Theano - 0.9.0rc3 ,keras - 1.2.2 ,pycuda - 2016.1.2 + cuda7518 ,gpu - Geforce GTX 480(其中2个)

Theano配置如下 .theanorc.txt

[global]
floatX = float32
device = gpu

[nvcc]
flags=-LC:\ProgramData\Anaconda2\libs
compiler_bindir=C:\Program Files (x86)\Microsoft Visual Studio 12.0\VC\bin

[lib]
cnmem=0.8

目前,我只能使用一个GPU,当我尝试安装模型时,我会收到以下内存错误

MemoryError:('错误分配411041792字节的设备内存(CNMEM_STATUS_OUT_OF_MEMORY)。',"您可以考虑使用' theano.shared(...,borrow = True) '&#34)

使用2 gpus解决问题(如果是,我如何启用第二个?) 或者我的模特太大了?

谢谢

0 个答案:

没有答案