File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
cscs-checks/apps/tensorflow Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 44
55class TensorFlowBaseTest (rfm .RunOnlyRegressionTest ):
66 def __init__ (self , model_name ):
7- super ().__init__ ()
87 self .name = 'tensorflow_%s_check' % model_name
98 self .descr = 'Tensorflow official %s test' % model_name
109 self .valid_systems = ['daint:gpu' , 'dom:gpu' ]
@@ -14,11 +13,14 @@ def __init__(self, model_name):
1413 self .tags = {'production' }
1514 self .num_tasks = 1
1615 self .num_gpus_per_node = 1
17- self .modules = ['TensorFlow/1.12.0-CrayGNU-19.03-cuda-10.0-python3' ]
16+ tf_version = '1.14.0'
17+ cuda_version = '10.1.168'
18+ self .modules = ['TensorFlow/%s-CrayGNU-19.06-cuda-%s-python3' %
19+ (tf_version , cuda_version )]
1820
1921 # Checkout to the branch corresponding to the module version of
2022 # TensorFlow
21- self .pre_run = ['git checkout r1.12.0' ]
23+ self .pre_run = ['git checkout r%s' % tf_version ]
2224 self .variables = {'PYTHONPATH' : '$PYTHONPATH:.' }
2325
2426
You can’t perform that action at this time.
0 commit comments