Update onnx and onnxruntime for torch tests (#2607)

### Changes

Bump up onnx, onnxruntime versions for torch tests
Update version of onnx at Readme

README.md

@@ -385,7 +385,7 @@ conda install -c conda-forge nncf
 - Supported frameworks:
   - PyTorch\* >=2.1, <2.3
   - TensorFlow\* >=2.8.4, <=2.12.1
-  - ONNX\* ~=1.13.1
+  - ONNX\* ==1.16.0
   - OpenVINO\* >=2022.3.0
 
 This repository is tested on Python* 3.8.10, PyTorch* 2.2.1 (NVidia CUDA\* Toolkit 12.1) and TensorFlow* 2.12.1 (NVidia CUDA\* Toolkit 11.8).

examples/post_training_quantization/onnx/mobilenet_v2/requirements.txt

@@ -2,6 +2,6 @@ torchvision
 tqdm
 scikit-learn
 fastdownload
-onnx~=1.16.0
-onnxruntime~=1.17.1
+onnx==1.16.0
+onnxruntime==1.17.1
 openvino-dev==2024.0

examples/post_training_quantization/onnx/yolov8_quantize_with_accuracy_control/requirements.txt

@@ -1,3 +1,3 @@
 ultralytics==8.0.170
-onnx~=1.16.0
+onnx==1.16.0
 openvino-dev==2024.0

examples/post_training_quantization/torch/ssd300_vgg16/requirements.txt

@@ -1,5 +1,5 @@
 fastdownload==0.0.7
-onnx==1.13.1
+onnx==1.16.0
 openvino-dev==2024.0
 pycocotools==2.0.7
 torch==2.2.1

setup.py

@@ -131,7 +131,7 @@ def find_version(*file_paths):
     "torch==2.2.1;python_version < '3.11'",
 ]
 
-ONNX_EXTRAS = ["onnx~=1.16.0", "onnxruntime~=1.17.1;python_version < '3.11'"]
+ONNX_EXTRAS = ["onnx==1.16.0", "onnxruntime==1.17.1;python_version < '3.11'"]
 
 OPENVINO_EXTRAS = ["openvino==2024.0"]
 

tests/post_training/requirements.txt

@@ -3,8 +3,8 @@ torch==2.2.1
 --extra-index-url https://download.pytorch.org/whl/cpu
 torchvision==0.17.1
 transformers==4.36.0
-onnx~=1.16.0
-onnxruntime~=1.17.1
+onnx==1.16.0
+onnxruntime==1.17.1
 pytest==8.0.2
 pytest-cov
 openvino-dev==2024.0.0

tests/torch/requirements.txt

@@ -1,7 +1,7 @@
 yattag>=1.14.0
 prettytable>=2.0.0
-onnx==1.13.1
-onnxruntime==1.14.1
+onnx==1.16.0
+onnxruntime==1.17.1
 pytest-mock>=3.3.1
 pytest-cov
 pytest-dependency>=0.5.1