# This is a minimal environment for running TTNN examples.
# Includes only the essential packages required to execute the prepared tutorials and examples.
# To install, run: ./tutorials_venv.sh

# General
numpy>=1.24.4,<2

# Torch packages
torch==2.2.1.0+cpu ; platform_machine == 'x86_64'
torch==2.2.1.0 ; platform_machine == 'aarch64'
torchvision==0.17.1+cpu ; platform_machine == 'x86_64'
torchvision==0.17.1 ; platform_machine == 'aarch64'
torchmetrics==1.3.1
torch-fidelity==0.3.0
