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

Added config variable to disable initial testing of nvidia drivers

上级 f4268206
...@@ -107,6 +107,7 @@ if config.device.startswith('gpu') or config.init_gpu_device.startswith('gpu'): ...@@ -107,6 +107,7 @@ if config.device.startswith('gpu') or config.init_gpu_device.startswith('gpu'):
if theano.sandbox.cuda.cuda_available: if theano.sandbox.cuda.cuda_available:
import theano.sandbox.cuda.tests.test_driver import theano.sandbox.cuda.tests.test_driver
if config.enable_initial_driver_test:
theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1() theano.sandbox.cuda.tests.test_driver.test_nvidia_driver1()
if (config.device.startswith('cuda') or if (config.device.startswith('cuda') or
......
...@@ -146,6 +146,13 @@ AddConfigVar( ...@@ -146,6 +146,13 @@ AddConfigVar(
in_c_key=False) 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(): def default_cuda_root():
v = os.getenv('CUDA_ROOT', "") v = os.getenv('CUDA_ROOT', "")
if v: if v:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论