提交 6fcc4739 authored 作者: Frederic's avatar Frederic

fix theano import crash on computer without gpu when theano flags device or…

fix theano import crash on computer without gpu when theano flags device or init_gpu_device start with gpu.
上级 45bc6488
......@@ -44,6 +44,9 @@ Internal changes
Crash Fix
* Don't try to use blas library when told to don't use them(Frederic B.)
* When importing theano on a computer without without gpu with the Theano
flags device or init_gpu_device to gpu* (Frederic B., Reported Luo Heng)
=============
Release Notes
......
......@@ -101,6 +101,7 @@ if config.device.startswith('gpu') or config.init_gpu_device.startswith('gpu'):
# We can't test the driver during import of theano.sandbox.cuda as
# this cause circular import dependency. So we also test it manually
# after the import
if theano.sandbox.cuda.cuda_available:
import theano.sandbox.cuda.tests.test_driver
theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论