| # CPU-side (x86) environment: data pipeline, normalization, unit tests. | |
| # GPU-side python comes from the training container ($SIF), not from here. | |
| --extra-index-url https://download.pytorch.org/whl/cpu | |
| torch==2.12.1+cpu | |
| numpy==2.5.0 | |
| pyarrow==24.0.0 | |
| pytest==9.1.1 | |
| huggingface_hub==1.22.0 | |
| transformers==4.57.6 | |
| safetensors==0.7.0 | |