提交 8ca002e9 authored 作者: joncrall's avatar joncrall

Added config variable to disable initial testing of nvidia drivers

上级 f4268206
......@@ -107,7 +107,8 @@ if config.device.startswith('gpu') or config.init_gpu_device.startswith('gpu'):
if theano.sandbox.cuda.cuda_available:
import theano.sandbox.cuda.tests.test_driver
theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()
if config.enable_initial_driver_test:
theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()
if (config.device.startswith('cuda') or
config.device.startswith('opencl') or
......
......@@ -146,6 +146,13 @@ AddConfigVar(
in_c_key=False)
AddConfigVar(
'enable_initial_driver_test',
"Tests the nvidia driver when a GPU device is initialized.",
BoolParam(True, allow_override=False),
in_c_key=False)
def default_cuda_root():
v = os.getenv('CUDA_ROOT', "")
if v:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论