使用带keras的cuda.use()修复“请切换到gpuarray后端”警告

Question

在使用时

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

这很棒，但是在网址上我看不到任何关于使用cuda.use()的东西。

我尝试了各种组合：theano.sandbox.cuda,use()、theano.gpuarray.use()、gpuarry.use()等，并用尽了组合的可能性。我们该怎么做？

Answer 1

删除cuda.use和theano.sandbox.cuda。将它替换为这个

import theano.gpuarray

现在转到您的.theanorc配置文件。如果使用linux并更改全局设置，则应在~/.theanorc中

device = cuda0

如果你找不到一个can文件，你可以搜索一下。