diff --git a/finetune/Dockerfile.conda1 b/finetune/Dockerfile.conda1 index 2ac311b..c94d7db 100644 --- a/finetune/Dockerfile.conda1 +++ b/finetune/Dockerfile.conda1 @@ -105,7 +105,7 @@ ENV _GLIBCXX_USE_CXX11_ABI=${_GLIBCXX_USE_CXX11_ABI} RUN <> ~/.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