提交 e913f6f0 authored 作者: Frederic's avatar Frederic

Fix import crash when nose isn't installed and we use the GPU back-end

上级 516bb822
...@@ -2,11 +2,15 @@ import numpy ...@@ -2,11 +2,15 @@ import numpy
import theano import theano
# Skip test if cuda_ndarray is not available. # Skip test if cuda_ndarray is not available.
from nose.plugins.skip import SkipTest try:
import theano.sandbox.cuda as cuda_ndarray from nose.plugins.skip import SkipTest
if cuda_ndarray.cuda_available == False: import theano.sandbox.cuda as cuda_ndarray
raise SkipTest('Optional package cuda disabled') if cuda_ndarray.cuda_available == False:
raise SkipTest('Optional package cuda disabled')
except ImportError:
# To have the GPU back-end work without nose, we need this file to
# be importable without nose.
pass
from theano.gof.python25 import any from theano.gof.python25 import any
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as cuda
import theano.sandbox.cuda.basic_ops as B import theano.sandbox.cuda.basic_ops as B
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论