@@ -52,27 +52,15 @@ The following prerequisites and requirements must be satisfied for a successful
5252 pip install -i https://test.pypi.org/simple/ neural-compressor
5353 ```
5454
55- - Install from Conda
56- ``` Shell
57- # install on Linux OS
58- conda install opencv-python-headless -c fastai
59- conda install neural-compressor -c conda-forge -c intel
60- ```
61- ``` Shell
62- # install on Windows OS
63- conda install pycocotools -c esri
64- conda install opencv-python-headless -c fastai
65- conda install neural-compressor -c conda-forge -c intel
66- ```
67-
6855### Install from Source
6956
7057 ``` Shell
7158 git clone https://github.com/intel/neural-compressor.git
7259 cd neural-compressor
7360 pip install -r requirements.txt
74- # build with basic functionality
7561 python setup.py install
62+ [optional] pip install requirements_pt.txt # for PyTorch framework extension API
63+ [optional] pip install requirements_tf.txt # for TensorFlow framework extension API
7664 ```
7765
7866### Install from AI Kit
@@ -112,7 +100,6 @@ The AI Kit is distributed through many common channels, including from Intel's w
112100 <tr style =" vertical-align : middle ; text-align : center ;" >
113101 <th>Framework</th>
114102 <th>TensorFlow</th>
115- <th>Intel<br>TensorFlow</th>
116103 <th>Intel®<br>Extension for<br>TensorFlow*</th>
117104 <th>PyTorch</th>
118105 <th>Intel®<br>Extension for<br>PyTorch*</th>
@@ -122,25 +109,26 @@ The AI Kit is distributed through many common channels, including from Intel's w
122109<tbody >
123110 <tr align =" center " >
124111 <th>Version</th>
125- <td class="tg-7zrl"> <a href=https://github.com/tensorflow/tensorflow/tree/v2.15.0>2.15.0</a><br>
126- <a href=https://github.com/tensorflow/tensorflow/tree/v2.14.1>2.14.1</a><br>
127- <a href=https://github.com/tensorflow/tensorflow/tree/v2.13.1>2.13.1</a><br></td>
128- <td class="tg-7zrl"> <a href=https://github.com/Intel-tensorflow/tensorflow/tree/v2.14.0>2.14.0</a><br>
129- <a href=https://github.com/Intel-tensorflow/tensorflow/tree/v2.13.0>2.13.0</a><br></td>
130- <td class="tg-7zrl"> <a href=https://github.com/intel/intel-extension-for-tensorflow/tree/v2.14.0.1>2.14.0.1</a><br>
112+ <td class="tg-7zrl">
113+ <a href=https://github.com/tensorflow/tensorflow/tree/v2.16.1>2.16.1</a><br>
114+ <a href=https://github.com/tensorflow/tensorflow/tree/v2.15.0>2.15.0</a><br>
115+ <a href=https://github.com/tensorflow/tensorflow/tree/v2.14.1>2.14.1</a><br></td>
116+ <td class="tg-7zrl">
117+ <a href=https://github.com/intel/intel-extension-for-tensorflow/tree/v2.15.0.0>2.15.0.0</a><br>
118+ <a href=https://github.com/intel/intel-extension-for-tensorflow/tree/v2.14.0.1>2.14.0.1</a><br>
131119 <a href=https://github.com/intel/intel-extension-for-tensorflow/tree/v2.13.0.0>2.13.0.0</a><br></td>
132- <td class="tg-7zrl"><a href=https://github.com/pytorch/pytorch/tree/v2.2.1>2.2.1</a><br>
133- <a href=https://github.com/pytorch/pytorch/tree/v2.1.0>2.1.0</a><br>
134- <a href=https://github.com/pytorch/pytorch/tree/v2.0.1>2.0.1</a><br></td>
135- <td class="tg-7zrl"><a href=https://github.com/intel/intel-extension-for-pytorch/tree/v2.2.0%2Bcpu>2.2.0</a><br>
136- <a href=https://github.com/intel/intel-extension-for-pytorch/tree/v2.1.100%2Bcpu>2.1.100</a><br>
137- <a href=https://github.com/intel/intel-extension-for-pytorch/tree/v2.0.100%2Bcpu>2.0.100</a><br></td>
138- <td class="tg-7zrl"><a href=https://github.com/microsoft/onnxruntime/tree/v1.17.1>1.17.1</a><br>
139- <a href=https://github.com/microsoft/onnxruntime/tree/v1.16.3>1.16.3</a><br>
140- <a href=https://github.com/microsoft/onnxruntime/tree/v1.15.1>1.15.1</a><br></td>
120+ <td class="tg-7zrl">
121+ <a href=https://github.com/pytorch/pytorch/tree/v2.3.0>2.3.0</a><br>
122+ <a href=https://github.com/pytorch/pytorch/tree/v2.2.2>2.2.2</a><br>
123+ <a href=https://github.com/pytorch/pytorch/tree/v2.1.1>2.1.1</a><br></td>
124+ <td class="tg-7zrl">
125+ <a href=https://github.com/intel/intel-extension-for-pytorch/tree/v2.3.0%2Bcpu>2.3.0</a><br>
126+ <a href=https://github.com/intel/intel-extension-for-pytorch/tree/v2.2.0%2Bcpu>2.2.0</a><br>
127+ <a href=https://github.com/intel/intel-extension-for-pytorch/tree/v2.1.100%2Bcpu>2.1.100</a><br></td>
128+ <td class="tg-7zrl">
129+ <a href=https://github.com/microsoft/onnxruntime/tree/v1.18.0>1.18.0</a><br>
130+ <a href=https://github.com/microsoft/onnxruntime/tree/v1.17.3>1.17.3</a><br>
131+ <a href=https://github.com/microsoft/onnxruntime/tree/v1.16.3>1.16.3</a><br></td>
141132 </tr >
142133</tbody >
143134</table >
144-
145- > ** Note:**
146- > Set the environment variable `` TF_ENABLE_ONEDNN_OPTS=1 `` to enable oneDNN optimizations if you are using TensorFlow before v2.9. oneDNN is the default for TensorFlow since [ v2.9] ( https://github.com/tensorflow/tensorflow/releases/tag/v2.9.0 ) ([ Intel Cascade Lake] ( https://www.intel.com/content/www/us/en/products/platforms/details/cascade-lake.html ) and newer CPUs).
0 commit comments