TPU
This commit is contained in:
40
model_training_nnn_tpu/requirements_tf.txt
Normal file
40
model_training_nnn_tpu/requirements_tf.txt
Normal file
@@ -0,0 +1,40 @@
|
||||
# TensorFlow Brain-to-Text Requirements for TPU v5e-8
|
||||
# Install with: pip install -r requirements_tf.txt
|
||||
|
||||
# Core TensorFlow and TPU support
|
||||
tensorflow>=2.15.0
|
||||
tensorflow-text>=2.15.0
|
||||
|
||||
# Data processing
|
||||
numpy>=1.21.0
|
||||
h5py>=3.7.0
|
||||
scipy>=1.9.0
|
||||
|
||||
# Configuration and utilities
|
||||
omegaconf>=2.3.0
|
||||
pyyaml>=6.0
|
||||
|
||||
# Logging and monitoring
|
||||
tensorboard>=2.12.0
|
||||
wandb>=0.15.0 # Optional: for experiment tracking
|
||||
|
||||
# Audio processing (for phoneme analysis)
|
||||
librosa>=0.10.0
|
||||
|
||||
# Development and testing
|
||||
pytest>=7.0.0
|
||||
pytest-cov>=4.0.0
|
||||
black>=22.0.0
|
||||
flake8>=5.0.0
|
||||
|
||||
# Optional: Jupyter for analysis notebooks
|
||||
jupyter>=1.0.0
|
||||
matplotlib>=3.5.0
|
||||
seaborn>=0.11.0
|
||||
pandas>=1.4.0
|
||||
|
||||
# Memory optimization
|
||||
psutil>=5.8.0
|
||||
|
||||
# Note: For TPU v5e-8 environments, TensorFlow should be pre-installed
|
||||
# with TPU support. This requirements file is for additional dependencies.
|
Reference in New Issue
Block a user