requirements.txt

openvino-dev[onnx,tensorflow2]==2021.4.*
matplotlib<3.4
gdown
pytube
yaspin

# ONNX notebook requirements
geffnet==0.9.8
fastseg
ipywidgets

torch>=1.5.1,<=1.7.1; sys_platform == 'darwin'
torchvision>=0.6.1,<=0.8.2; sys_platform == 'darwin'
--find-links https://download.pytorch.org/whl/torch_stable.html
torch>=1.5.1+cpu,<=1.7.1+cpu; sys_platform =='linux' or platform_system == 'Windows'
torchvision>=0.6.1+cpu,<=0.8.2+cpu; sys_platform =='linux' or platform_system == 'Windows'

# PaddlePaddle notebook requirements
# For 103 PaddlePaddle MO conversion tutorial and 206 PaddleGAN/AnimeGAN demo
paddlepaddle==2.1.*
paddlehub
paddle2onnx>=0.6
ppgan>=2.0.0

# BERT quantization notebook requirements
transformers
sklearn

# Jupyter requirements
jupyterlab

# Pin versions to prevent known dependency issues
ipython==7.10.*
jedi==0.17.2
setuptools>=56.0.0
Pillow>=8.3.2
ipykernel==5.*
pygments>=2.7.4 # not directly required, pinned by Snyk to avoid a vulnerability
nltk>=3.6.4 # not directly required, pinned by Snyk to avoid a vulnerability
rsa>=4.7 # not directly required, pinned by Snyk to avoid a vulnerability
scikit-learn>=0.24.2 # not directly required, pinned by Snyk to avoid a vulnerability

# NNCF notebook requirements
nncf[torch]
nncf[tf]
tensorflow_datasets==4.2.0