ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix torch-npu dependency

Browse Source 
Former-commit-id: 7c8a8061d0cda6342f6c883748fb6bc6650df9f9
This commit is contained in:
fanjunliang
2024-06-26 18:21:42 +08:00
parent bde31d8600
commit d82d86e16d
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
docker/docker-npu/Dockerfile
View File

@@ -7,16 +7,17 @@ ENV DEBIAN_FRONTEND=noninteractive
# Define installation arguments
ARG INSTALL_DEEPSPEED=false
ARG PIP_INDEX=https://pypi.org/simple
ARG EXTRA_INDEX=https://download.pytorch.org/whl/cpu
# Set the working directory
WORKDIR /app
# Install the requirements
COPY requirements.txt /app
RUN pip config set global.index-url $PIP_INDEX
RUN pip config set global.extra-index-url $PIP_INDEX
RUN python -m pip install --upgrade pip
RUN python -m pip install -r requirements.txt
RUN pip config set global.index-url $PIP_INDEX && \
pip config set global.extra-index-url $EXTRA_INDEX && \
pip install --upgrade pip && \
pip install -r requirements.txt
# Copy the rest of the application into the image
COPY . /app