lingyuzeng/cdc_dockerfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove cmake

Browse Source
This commit is contained in:
hotwa
2024-06-21 12:01:51 +08:00
parent c69bbe2c71
commit 5b4c0ca74d
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
finetune/Dockerfile.conda1
View File

@@ -105,7 +105,7 @@ ENV _GLIBCXX_USE_CXX11_ABI=${_GLIBCXX_USE_CXX11_ABI}
RUN <<EOT
#!/bin/bash
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
conda activate ${CONDA_ENV_NAME}
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)
python3 -m pip uninstall -y torch torchvision torchaudio
# # install pytorch create conda env aleay exists
conda install -y cmake ninja
git clone --recursive https://github.com/pytorch/pytorch ${STAGE_DIR}/pytorch
cd ${STAGE_DIR}/pytorch
git submodule sync