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

add --no-build-isolation flash-attn

Browse Source
This commit is contained in:
hotwa
2024-06-15 17:53:29 +08:00
parent d82aa8a65d
commit f2e426fddd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/Dockerfile
View File

@@ -190,7 +190,7 @@ RUN <<EOT
source /opt/conda/etc/profile.d/conda.sh source /opt/conda/etc/profile.d/conda.sh
conda activate ${CONDA_ENV_NAME} conda activate ${CONDA_ENV_NAME}
pip install torch==${PYTORCH_VERSION} torchvision==0.17.1 torchaudio==${PYTORCH_VERSION} --index-url https://download.pytorch.org/whl/cu121 pip install torch==${PYTORCH_VERSION} torchvision==0.17.1 torchaudio==${PYTORCH_VERSION} --index-url https://download.pytorch.org/whl/cu121
pip install --no-build-isolation flash-attn -i https://pypi.org/simple/ --trusted-host pypi.org pip install flash-attn --no-build-isolation -i https://pypi.org/simple/ --trusted-host pypi.org
pip install deepspeed bitsandbytes accelerate transformers optimum pip install deepspeed bitsandbytes accelerate transformers optimum
pip install pydantic transformers datasets accelerate evaluate peft deepspeed tiktoken \ pip install pydantic transformers datasets accelerate evaluate peft deepspeed tiktoken \
sentencepiece tqdm nltk matplotlib seaborn numpy pandas scikit-learn diffusers \ sentencepiece tqdm nltk matplotlib seaborn numpy pandas scikit-learn diffusers \