refactor: 尝试去掉nvidia依赖

youkaede77 · Mar 21, 2024 · 6533b87 · 6533b87
1 parent d1e4ba1
commit 6533b87
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/installer/Dockerfile b/installer/Dockerfile
@@ -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/*