提交 4d93ae26 authored 作者: Frederic's avatar Frederic

disabled pycuda_init.py in Theano as this make the tests fails.

This is probably due to wrong handling of context. I think theano and pycuda don't use the same context while the should. I am investigating the problem, but while this make the buildbot segfault, I disable it for now.
上级 368b6d65
...@@ -24,7 +24,7 @@ def set_gpu_from_theano(): ...@@ -24,7 +24,7 @@ def set_gpu_from_theano():
set_gpu_from_theano() set_gpu_from_theano()
pycuda_available = False pycuda_available = False
if True: # theano.sandbox.cuda.use.device_number is None: if False:
try: try:
import pycuda import pycuda
import pycuda.autoinit import pycuda.autoinit
...@@ -33,6 +33,3 @@ if True: # theano.sandbox.cuda.use.device_number is None: ...@@ -33,6 +33,3 @@ if True: # theano.sandbox.cuda.use.device_number is None:
# presumably, the user wanted to use pycuda, else they wouldn't have # presumably, the user wanted to use pycuda, else they wouldn't have
# imported this module, so issue a warning that the import failed. # imported this module, so issue a warning that the import failed.
warnings.warn("PyCUDA import failed in theano.misc.pycuda_init") warnings.warn("PyCUDA import failed in theano.misc.pycuda_init")
else:
warnings.warn("theano.misc.pycuda_init must be imported before theano"
" init its GPU")
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论