File size: 327 Bytes
5952424
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
# 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