提交 6073756f authored 作者: Frederic's avatar Frederic

Fix important crash. This make many failure in the daily buildbot

上级 857e7dc5
import pkg_resources
import theano import theano
from theano.sandbox.cuda.type import CudaNdarrayType from theano.sandbox.cuda.type import CudaNdarrayType
from theano.sandbox.cuda import GpuOp from theano.sandbox.cuda import GpuOp
from theano.sandbox.cuda.basic_ops import as_cuda_ndarray_variable from theano.sandbox.cuda.basic_ops import as_cuda_ndarray_variable
try: try:
...@@ -15,7 +16,7 @@ cula_available = False ...@@ -15,7 +16,7 @@ cula_available = False
try: try:
from scikits.cuda import cula from scikits.cuda import cula
cula_available = True cula_available = True
except (ImportError, OSError): except (ImportError, OSError, pkg_resources.DistributionNotFound):
pass pass
cula_initialized = False cula_initialized = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论