2bf8d910f6
* [Docs][PyOV] update python snippets * first snippet * Fix samples debug * Fix linter * part1 * Fix speech sample * update model state snippet * add serialize * add temp dir * CPU snippets update (#134) * snippets CPU 1/6 * snippets CPU 2/6 * snippets CPU 3/6 * snippets CPU 4/6 * snippets CPU 5/6 * snippets CPU 6/6 * make module TODO: REMEMBER ABOUT EXPORTING PYTONPATH ON CIs ETC * Add static model creation in snippets for CPU * export_comp_model done * leftovers * apply comments * apply comments -- properties * small fixes * rempve debug info * return IENetwork instead of Function * apply comments * revert precision change in common snippets * update opset * [PyOV] Edit docs for the rest of plugins (#136) * modify main.py * GNA snippets * GPU snippets * AUTO snippets * MULTI snippets * HETERO snippets * Added properties * update gna * more samples * Update docs/OV_Runtime_UG/model_state_intro.md * Update docs/OV_Runtime_UG/model_state_intro.md * attempt1 fix ci * new approach to test * temporary remove some files from run * revert cmake changes * fix ci * fix snippet * fix py_exclusive snippet * fix preprocessing snippet * clean-up main * remove numpy installation in gha * check for GPU * add logger * iexclude main * main update * temp * Temp2 * Temp2 * temp * Revert temp * add property execution devices * hide output from samples --------- Co-authored-by: p-wysocki <przemyslaw.wysocki@intel.com> Co-authored-by: Jan Iwaszkiewicz <jan.iwaszkiewicz@intel.com> Co-authored-by: Karol Blaszczak <karol.blaszczak@intel.com>
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
# Copyright (C) 2022 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
|
|
import openvino as ov
|
|
from snippets import get_model
|
|
|
|
|
|
def main():
|
|
model = get_model()
|
|
|
|
core = ov.Core()
|
|
if "GPU" not in core.available_devices:
|
|
return 0
|
|
|
|
#! [compile_model_default_gpu]
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model(model, "GPU")
|
|
#! [compile_model_default_gpu]
|
|
|
|
#! [compile_model_gpu_with_id]
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model(model, "GPU.1")
|
|
#! [compile_model_gpu_with_id]
|
|
|
|
#! [compile_model_gpu_with_id_and_tile]
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model(model, "GPU.1.0")
|
|
#! [compile_model_gpu_with_id_and_tile]
|
|
|
|
#! [compile_model_multi]
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model(model, "MULTI:GPU.1,GPU.0")
|
|
#! [compile_model_multi]
|
|
|
|
#! [compile_model_batch_plugin]
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model(model, "BATCH:GPU")
|
|
#! [compile_model_batch_plugin]
|
|
|
|
#! [compile_model_auto_batch]
|
|
core = ov.Core()
|
|
compiled_model = core.compile_model(
|
|
model,
|
|
"GPU",
|
|
{
|
|
ov.properties.hint.performance_mode(): ov.properties.hint.PerformanceMode.THROUGHPUT,
|
|
},
|
|
)
|
|
#! [compile_model_auto_batch]
|