Use porto layer with nvidia-396 driver to fix catboost tests
issue:https://st.yandex-team.ru/DEVTOOLS-4881
@@ -78,8 +78,18 @@ def execute(*args, **kwargs):
pytest.xfail(reason=cuda_setup_error)
return yatest.yt.execute(
*args,
- task_spec={'gpu_limit': 1},
- operation_spec={'pool_trees': ['gpu']},
+ task_spec={
+ # temporary layers
+ 'layer_paths': [
+ '//home/codecoverage/nvidia-396.tar.gz',
+ '//porto_layers/ubuntu-xenial-base.tar.xz',
+ ],
+ 'gpu_limit': 1
+ },
+ operation_spec={
+ 'pool_trees': ['gpu_geforce_1080ti'],
+ 'scheduling_tag_filter': 'porto',
+ },
input_data=input_data,
output_data=output_data,
# required for quantized-marked input filenames
