1

我是深度学习/theano/keras 的初学者。我正在尝试弄清楚如何在 Windows 7 上使用多个 GPU。我已经成功安装了 Theano、keras(如本文中所述如何安装 Keras 和 Theano在 Windows 上的 Anaconda Python 中?)并使用一个 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: ('Error allocating 411041792 bytes of device memory (CNMEM_STATUS_OUT_OF_MEMORY).', "你可以考虑使用'theano.shared(..., borrow=True)'")

使用 2 gpus 是否可以解决问题(如果是,如何启用第二个?)还是我的模型太大?

谢谢你

4

0 回答 0