提交 a6f9131b authored 作者: Pascal Lamblin's avatar Pascal Lamblin 提交者: GitHub

Merge pull request #5946 from nouiz/better_error

Give a good error when people use device=gpu
...@@ -93,11 +93,17 @@ class DeviceParam(ConfigParam): ...@@ -93,11 +93,17 @@ class DeviceParam(ConfigParam):
val.startswith('opencl') or val.startswith('opencl') or
val.startswith('cuda')): val.startswith('cuda')):
return val return val
elif val.startswith('gpu'):
raise ValueError(
'You are tring to use the old GPU back-end. '
'It was removed from Theano. Use device=cuda* now. '
'See https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29 '
'for more information.')
else: else:
raise ValueError(('Invalid value ("%s") for configuration ' raise ValueError(('Invalid value ("%s") for configuration '
'variable "%s". Valid options start with ' 'variable "%s". Valid options start with '
'one of "%s", "opencl", "cuda"' 'one of "cpu", "opencl" or "cuda".'
% (self.default, val, self.fullname))) % (val, self.fullname)))
over = kwargs.get("allow_override", True) over = kwargs.get("allow_override", True)
super(DeviceParam, self).__init__(default, filter, over) super(DeviceParam, self).__init__(default, filter, over)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论