提交 dfa86140 authored 作者: James Bergstra's avatar James Bergstra

moved nvcc_compiler import in cuda init.py

上级 439b6964
......@@ -4,6 +4,7 @@ import logging
from theano.compile import optdb
from theano import config
from theano.gof.cmodule import get_lib_extension
import nvcc_compiler
_logger_name = 'theano.sandbox.cuda'
_logger = logging.getLogger(_logger_name)
......@@ -86,7 +87,6 @@ if not compile_cuda_ndarray:
try:
if compile_cuda_ndarray:
import nvcc_compiler
if not nvcc_compiler.is_nvcc_available():
# Hide the error message if the user manually unset the
# config.cuda.root configuration variable.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论