提交 82757a69 authored 作者: Frederic's avatar Frederic

Fix test when pygpu isn't installed.

上级 e10042c5
......@@ -12,6 +12,9 @@ from numpy.testing.noseclasses import KnownFailureTest
import theano.sandbox.gpuarray
if theano.sandbox.gpuarray.pygpu is None:
raise SkipTest("pygpu not installed")
import theano.sandbox.cuda as cuda_ndarray
if cuda_ndarray.cuda_available and not theano.sandbox.gpuarray.pygpu_activated:
if not cuda_ndarray.use.device_number:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论