From b4b5125b01bde5679a0110e31d4ecef8e2514d6e Mon Sep 17 00:00:00 2001 From: XinyuYe-Intel Date: Tue, 17 Dec 2024 11:27:01 +0800 Subject: [PATCH] Update Dockerfile --- comps/finetuning/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comps/finetuning/Dockerfile b/comps/finetuning/Dockerfile index afcf36f15..eca81f9cd 100644 --- a/comps/finetuning/Dockerfile +++ b/comps/finetuning/Dockerfile @@ -23,7 +23,7 @@ ENV PATH=$PATH:/home/user/.local/bin RUN python -m pip install --no-cache-dir --upgrade pip && \ python -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu && \ python -m pip install --no-cache-dir intel-extension-for-pytorch && \ - python -m pip install --no-cache-dir oneccl_bind_pt --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/ && \ + python -m pip install --no-cache-dir oneccl_bind_pt --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/cn/ && \ python -m pip install --no-cache-dir -r /home/user/comps/finetuning/requirements.txt ENV PYTHONPATH=$PYTHONPATH:/home/user