File size: 208 Bytes
b16c3a6
 
 
 
 
1
2
3
4
5
6
# shared runtime env for training/eval on the shared-CPU B200 box
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export TOKENIZERS_PARALLELISM=false
export OMP_NUM_THREADS=4
export MKL_NUM_THREADS=4