提交 1814576c authored 作者: vesis84's avatar vesis84

force-disable of GPU initialization with pycuda

- pycuda initialization was also affected by the bug from CUDA 7.0
上级 eed9d97d
...@@ -399,9 +399,9 @@ def use(device, ...@@ -399,9 +399,9 @@ def use(device,
# Has PyCUDA already initialized the GPU context # Has PyCUDA already initialized the GPU context
pycuda_init_dev = False pycuda_init_dev = False
if config.pycuda.init: # if config.pycuda.init:
import theano.misc.pycuda_init # import theano.misc.pycuda_init
pycuda_init_dev = theano.misc.pycuda_init.pycuda_available # pycuda_init_dev = theano.misc.pycuda_init.pycuda_available
try: try:
if (device != 'gpu') and not pycuda_init_dev: if (device != 'gpu') and not pycuda_init_dev:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论