remove cmake

This commit is contained in:
2024-06-21 12:01:51 +08:00
parent c69bbe2c71
commit 5b4c0ca74d

View File

@@ -105,7 +105,7 @@ ENV _GLIBCXX_USE_CXX11_ABI=${_GLIBCXX_USE_CXX11_ABI}
RUN <<EOT RUN <<EOT
#!/bin/bash #!/bin/bash
source /opt/conda/etc/profile.d/conda.sh source /opt/conda/etc/profile.d/conda.sh
conda create -n ${CONDA_ENV_NAME} python=${PYTHON_VERSION} pyyaml ipython -c conda-forge -y conda create -n ${CONDA_ENV_NAME} python=${PYTHON_VERSION} pyyaml ipython cmake ninja -c conda-forge -y
echo "conda activate ${CONDA_ENV_NAME}" >> ~/.bashrc echo "conda activate ${CONDA_ENV_NAME}" >> ~/.bashrc
conda activate ${CONDA_ENV_NAME} conda activate ${CONDA_ENV_NAME}
python3 -m pip install --no-cache-dir --upgrade pip python3 -m pip install --no-cache-dir --upgrade pip
@@ -118,7 +118,6 @@ python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing]
# # (https://www.deepspeed.ai/tutorials/advanced-install/#pre-install-deepspeed-ops) # # (https://www.deepspeed.ai/tutorials/advanced-install/#pre-install-deepspeed-ops)
python3 -m pip uninstall -y torch torchvision torchaudio python3 -m pip uninstall -y torch torchvision torchaudio
# # install pytorch create conda env aleay exists # # install pytorch create conda env aleay exists
conda install -y cmake ninja
git clone --recursive https://github.com/pytorch/pytorch ${STAGE_DIR}/pytorch git clone --recursive https://github.com/pytorch/pytorch ${STAGE_DIR}/pytorch
cd ${STAGE_DIR}/pytorch cd ${STAGE_DIR}/pytorch
git submodule sync git submodule sync