dmlc / keras

Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on MXNet, Theano or TensorFlow.
http://keras.io/
Other
125 stars 34 forks source link

Error with Theano backend and multiple GPUs #78

Closed beeva-juangomez closed 7 years ago

beeva-juangomez commented 7 years ago

I'm trying to use Theano backend with multiple GPUs and I'm getting this error: ValueError: Invalid argument "context" passed to K.function. However, if i'm running TheanoGPU test script everything is correct. Any idea?

Thanks!

sandeep-krishnamurthy commented 7 years ago

Hello, Multi-GPUs via context parameter is the new functionality added only for MXNet backend. Theano backend does not support this functionality.

beeva-juangomez commented 7 years ago

Thank you, closing the issue