refactor: 尝试去掉nvidia依赖

This commit is contained in:
liqiang-fit2cloud 2024-03-21 15:08:27 +08:00
parent d1e4ba1c3f
commit 6533b87109

View File

@ -23,9 +23,12 @@ RUN mv /opt/maxkb/app/model/* /opt/maxkb/model && \
cp -r /opt/maxkb/model/base/hub /opt/maxkb/model/tokenizer && \
apt-get update && apt-get install -y curl && \
pip3 install --upgrade pip && \
pip3 install poetry && \
cat pyproject.toml && \
if [ `uname -m` == "aarch64" ]; then sed -i '/^torch/d' pyproject.toml; fi && \
cat pyproject.toml && \
pip3 install poetry && \
poetry export -f requirements.txt --output requirements.txt --without-hashes && \
cat requirements.txt && \
pip3 install --no-cache-dir -r requirements.txt && \
pip3 cache purge && \
rm -rf /var/lib/apt/lists/*