# 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.11.0 ; platform_machine == 'aarch64'
torchvision==0.26.0 ; platform_machine == 'aarch64'
torchmetrics==1.9.0
torch-fidelity==0.3.0

# CPU-only torch for x86_64 to avoid installing CUDA dependencies on dev machines
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.11.0 ; platform_machine == 'x86_64'
torchvision==0.26.0 ; platform_machine == 'x86_64'
