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

disable the CUDA back-end if nvcc is not there even if we can load cuda_ndarray.so

上级 80ded7a8
...@@ -163,6 +163,13 @@ if compile_cuda_ndarray: ...@@ -163,6 +163,13 @@ if compile_cuda_ndarray:
set_cuda_disabled() set_cuda_disabled()
finally: finally:
release_lock() release_lock()
elif not nvcc_compiler.is_nvcc_available():
# This can happen if there is cuda_ndarray.so was already compiled
# and then nvcc is removed. In that case we need to disable the CUDA
# back-end as we won't be able to compile any new op and we can't only
# use already compiled GPU op and not the others.
set_cuda_disabled()
del compile_cuda_ndarray del compile_cuda_ndarray
if cuda_available: if cuda_available:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论