修复'请切换到gpuarray后端'用kera

时间:2017-06-28 02:33:12

标签: keras theano

使用时

cuda.use('gpu0')

我得到了

WARNING (theano.sandbox.cuda): The cuda backend is deprecated and will be removed in the next release (v0.10). Please switch to the gpuarray backend. You can get more information about how to switch at this URL: https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29

哪个好,但在URL上没有任何关于使用cuda.use()的信息。

我尝试了各种组合:theano.sandbox.cuda,use()theano.gpuarray.use()gpuarry.use()等等,并且耗尽了组合的可能性。我们应该做什么?

1 个答案:

答案 0 :(得分:1)

删除cuda.use和theano.sandbox.cuda。用这个

替换它
import theano.gpuarray

现在转到.theanorc配置文件。如果linux和更改全局设置应该在〜/ .theanorc

device = cuda0
如果你找不到theanorc文件,谷歌周围。