remove cmake
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user