提交 27a4b85f authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make gpuarray actually usable.

上级 9aa5cec7
......@@ -81,6 +81,10 @@ from theano.updates import Updates, OrderedUpdates
from theano.gradient import Rop, Lop, grad
if config.device.startswith('cuda') or config.device.startswith('opencl') or \
config.gpuarray.init_device != '':
import theano.sandbox.gpuarray
if config.device.startswith('gpu') or config.init_gpu_device.startswith('gpu'):
import theano.sandbox.cuda
# We can't test the driver during import of theano.sandbox.cuda as
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论