@@ -52,27 +52,15 @@ The following prerequisites and requirements must be satisfied for a successful
52
52
pip install -i https://test.pypi.org/simple/ neural-compressor
53
53
```
54
54
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
-
68
55
### Install from Source
69
56
70
57
``` Shell
71
58
git clone https://github.com/intel/neural-compressor.git
72
59
cd neural-compressor
73
60
pip install -r requirements.txt
74
- # build with basic functionality
75
61
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
76
64
```
77
65
78
66
### Install from AI Kit
@@ -112,7 +100,6 @@ The AI Kit is distributed through many common channels, including from Intel's w
112
100
<tr style =" vertical-align : middle ; text-align : center ;" >
113
101
<th>Framework</th>
114
102
<th>TensorFlow</th>
115
- <th>Intel<br>TensorFlow</th>
116
103
<th>Intel®<br>Extension for<br>TensorFlow*</th>
117
104
<th>PyTorch</th>
118
105
<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
122
109
<tbody >
123
110
<tr align =" center " >
124
111
<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>
131
119
<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>
141
132
</tr >
142
133
</tbody >
143
134
</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