Skip to content

Commit c66b4f5

Browse files
committed
changes from #787
1 parent 90d56b2 commit c66b4f5

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

scripts/empireai_nvl72_train.sbatch

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,9 @@ export MKL_NUM_THREADS="${MKL_NUM_THREADS:-${OMP_NUM_THREADS}}"
118118
export OPENBLAS_NUM_THREADS="${OPENBLAS_NUM_THREADS:-${OMP_NUM_THREADS}}"
119119
export NUMEXPR_NUM_THREADS="${NUMEXPR_NUM_THREADS:-${OMP_NUM_THREADS}}"
120120
export PYTHONUNBUFFERED=1
121+
export PYTHONNOUSERSITE="${PYTHONNOUSERSITE:-1}"
122+
export APPTAINERENV_PYTHONNOUSERSITE="${APPTAINERENV_PYTHONNOUSERSITE:-${PYTHONNOUSERSITE}}"
123+
export SINGULARITYENV_PYTHONNOUSERSITE="${SINGULARITYENV_PYTHONNOUSERSITE:-${PYTHONNOUSERSITE}}"
121124
export NCCL_DEBUG="${NCCL_DEBUG:-WARN}"
122125
export NCCL_SOCKET_IFNAME="${NCCL_SOCKET_IFNAME:-bond0}"
123126
export NCCL_IB_HCA="${NCCL_IB_HCA:-mlx5_0,mlx5_1,mlx5_4,mlx5_5}"

0 commit comments

Comments
 (0)