From 4a6697ef699e4c02e5bbb14ff3a9a97917a9a40a Mon Sep 17 00:00:00 2001
From: lingyuzeng <pylyzeng@gmail.com>
Date: Thu, 18 Jul 2024 10:19:53 +0800
Subject: [PATCH] add env stage_dir at start

---
 spawnerdockerfile/Dockerfile.ngc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spawnerdockerfile/Dockerfile.ngc b/spawnerdockerfile/Dockerfile.ngc
index 73fdfc6..857b36c 100644
--- a/spawnerdockerfile/Dockerfile.ngc
+++ b/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