提交 487aa554 authored 作者: nouiz's avatar nouiz

Merge pull request #1275 from nouiz/fix_force_device_cpu

Fix PR gh-90
......@@ -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.
# Also, if cxx is not available, we need to disable all GPU code.
set_cuda_disabled()
compile_cuda_ndarray = False
elif not config.device.startswith('gpu') and config.force_device:
# We where asked to NEVER use the GPU
set_cuda_disabled()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论