samples/python/benchmark/bert_benhcmark/requirements.txt: add datasets, install torch cpu only for linux (#14765)

* samples/python/benchmark/bert_benhcmark/requirements.txt: add datasets, install torch cpu only for linux

By default for linux torch is installed with GPU support, but for
windows is CPU only

* benhcmark->benchmark
This commit is contained in:
Zlobin Vladimir authored and GitHub committed 2022-12-23 16:58:32 +04:00
1 parent fa61aed443
commit ecfeda32e3
5 files changed
+7 -5

No files matched your search

@@ -13,7 +13,7 @@ The following Python\* API is used in the application:
## How It Works
The sample downloads a model and a tokenizer, export the model to onnx, reads the exported model and reshapes it to enforce dynamic inpus shapes, compiles the resulting model, downloads a dataset and runs benhcmarking on the dataset.
The sample downloads a model and a tokenizer, export the model to onnx, reads the exported model and reshapes it to enforce dynamic inpus shapes, compiles the resulting model, downloads a dataset and runs benchmarking on the dataset.
You can see the explicit description of
each sample step at [Integration Steps](../../../../docs/OV_Runtime_UG/integrate_with_your_application.md) section of "Integrate OpenVINO™ Runtime with Your Application" guide.
@@ -35,7 +35,7 @@ python -m pip install -r requirements.txt
Run the sample
```
python bert_benhcmark.py
python bert_benchmark.py
```
## Sample Output
@@ -0,0 +1,4 @@
datasets
transformers[onnx]
--extra-index-url https://download.pytorch.org/whl/cpu
torch
@@ -1,2 +0,0 @@
transformers[onnx]
torch