add env stage_dir at start

spawnerdockerfile/Dockerfile.ngc

@@ -17,6 +17,7 @@ SHELL ["/bin/bash", "-c"]
 
 # https://network.nvidia.com/products/infiniband-drivers/linux/mlnx_ofed/
 ENV MLNX_OFED_VERSION=23.10-3.2.2.0 
+ENV STAGE_DIR=/tmp
 RUN <<EOT
 #!/bin/bash
 # SYSTEM_NAME=$(lsb_release -cs) # 查看发行版本
@@ -86,7 +87,6 @@ pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=
 pip install git+https://github.com/huggingface/transformers
 EOT
 
-ENV STAGE_DIR=/tmp
 RUN <<EOT
 #!/bin/bash
 git clone https://github.com/microsoft/DeepSpeed-Kernels.git ${STAGE_DIR}/DeepSpeed-Kernels