[assets] fix npu docker (#8298)

This commit is contained in:
Yaowei Zheng
2025-06-05 19:09:20 +08:00
committed by GitHub
parent 1a33d65a56
commit ed70f8d5a2
5 changed files with 34 additions and 30 deletions

View File

@@ -28,11 +28,11 @@ WORKDIR /app
# Change pip source
RUN pip config set global.index-url "${PIP_INDEX}" && \
pip config set global.extra-index-url "${PIP_INDEX}" && \
python -m pip install --upgrade pip
pip install --no-cache-dir --upgrade pip packaging wheel setuptools
# Reinstall pytorch rocm
RUN pip uninstall -y torch torchvision torchaudio && \
pip install --pre torch torchvision torchaudio --index-url "${PYTORCH_INDEX}"
pip install --no-cache-dir --pre torch torchvision torchaudio --index-url "${PYTORCH_INDEX}"
# Install the requirements
COPY requirements.txt /app