numpy>=1.21.0
psutil>=5.8.0
pyyaml>=6.0
torch>=2.0.0

[all]
nvidia-ml-py>=11.450.51
tensorrt>=8.0.0
pycuda>=2022.1
intel-extension-for-pytorch>=2.0.0
docker>=6.0.0

[amd]

[dev]
pytest>=7.0
black>=22.0
flake8>=4.0
mypy>=0.900
build>=0.10.0
twine>=4.0.2

[intel]
intel-extension-for-pytorch>=2.0.0

[nvidia]
nvidia-ml-py>=11.450.51
tensorrt>=8.0.0
pycuda>=2022.1
