lingyuzeng/docker-jupyterhub
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add env stage_dir at start

Browse Source
This commit is contained in:
lingyuzeng
2024-07-18 10:19:53 +08:00
parent 305fd85088
commit 4a6697ef69
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spawnerdockerfile/Dockerfile.ngc
View File

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