提交 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,
# Has PyCUDA already initialized the GPU context
pycuda_init_dev = False
if config.pycuda.init:
import theano.misc.pycuda_init
pycuda_init_dev = theano.misc.pycuda_init.pycuda_available
# if config.pycuda.init:
# import theano.misc.pycuda_init
# pycuda_init_dev = theano.misc.pycuda_init.pycuda_available
try:
if (device != 'gpu') and not pycuda_init_dev:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论