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:
1 parent
fa61aed443
commit
ecfeda32e3
5 files changed
+7
-5
No files matched your search
+2
-2
@@ -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
|
||||
File renamed without changes.
@@ -0,0 +1,4 @@
|
||||
datasets
|
||||
transformers[onnx]
|
||||
--extra-index-url https://download.pytorch.org/whl/cpu
|
||||
torch
|
||||
@@ -1,2 +0,0 @@
|
||||
transformers[onnx]
|
||||
torch
|
||||
Reference in new issue
Block a user