LLM/pentestagent
1
0
Fork 0
mirror of https://github.com/GH05TCREW/pentestagent.git synced 2026-03-08 06:44:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: split pip install for CI memory limits, use CPU torch

Browse Source
This commit is contained in:
GH05TCREW
2025-12-07 13:48:20 -07:00
parent ee7247d9fd
commit e1fd7c4f09
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile.kali
View File

@@ -57,8 +57,12 @@ WORKDIR /app
# Install Python dependencies using virtual environment (PEP 668 compliance)
COPY requirements.txt .
RUN python3 -m venv /opt/venv && \
/opt/venv/bin/pip install --no-cache-dir --upgrade pip && \
/opt/venv/bin/pip install --no-cache-dir -r requirements.txt
/opt/venv/bin/pip install --no-cache-dir --upgrade pip wheel setuptools
# Install heavy dependencies first (helps with CI memory limits)
RUN /opt/venv/bin/pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
RUN /opt/venv/bin/pip install --no-cache-dir sentence-transformers faiss-cpu
RUN /opt/venv/bin/pip install --no-cache-dir -r requirements.txt
# Add venv to PATH
ENV PATH="/opt/venv/bin:$PATH"