提交 0d78c703 authored 作者: Frederic's avatar Frederic

Fix PR gh-90

If cause import crash in some cases.
上级 44144c83
...@@ -121,6 +121,7 @@ if not nvcc_compiler.is_nvcc_available() or not theano.config.cxx: ...@@ -121,6 +121,7 @@ if not nvcc_compiler.is_nvcc_available() or not theano.config.cxx:
# use already compiled GPU op and not the others. # use already compiled GPU op and not the others.
# Also, if cxx is not available, we need to disable all GPU code. # Also, if cxx is not available, we need to disable all GPU code.
set_cuda_disabled() set_cuda_disabled()
compile_cuda_ndarray = False
elif not config.device.startswith('gpu') and config.force_device: elif not config.device.startswith('gpu') and config.force_device:
# We where asked to NEVER use the GPU # We where asked to NEVER use the GPU
set_cuda_disabled() set_cuda_disabled()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论