deepseekmirror/DeepSeek-Prover-V1.5
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Prover-V1.5.git synced 2025-07-03 14:41:37 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

docker env

Browse Source
This commit is contained in:
ondanaoto 2024-10-08 06:39:24 +00:00
parent 2e26a8c840
commit 64463209a9
2 changed files with 13 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
Dockerfile
View File

@ -1,34 +1,22 @@
# Use the official Python 3.10 image as base FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-devel
FROM python:3.10-slim
# Set environment variables # Set environment variables
ENV PYTHONUNBUFFERED=1
WORKDIR /app WORKDIR /app
# Install required system packages # Install required system packages and Python
RUN apt-get update && apt-get install -y \ RUN apt-get update -y && apt-get install -y \
curl \ curl \
wget \ git
git \ # Custom Lean 4 installation script without Visual Studio Code
sudo \ RUN curl -y https://raw.githubusercontent.com/leanprover/elan/master/elan-init.sh -sSf | sh
&& rm -rf /var/lib/apt/lists/*
# Install Lean 4 # Copy the requirements.txt file
RUN wget -q https://raw.githubusercontent.com/leanprover-community/mathlib4/master/scripts/install_debian.sh \
&& bash install_debian.sh \
&& rm -f install_debian.sh \
&& source ~/.profile
# Add lean to the system's PATH
ENV PATH="/root/.elan/bin:$PATH"
# Install Python dependencies (assumes that the requirements.txt is in the repository)
COPY requirements.txt /app/ COPY requirements.txt /app/
RUN pip install --no-cache-dir -r requirements.txt RUN pip3 install --no-cache-dir -r requirements.txt
# Build Mathlib4 during the image build # Build Mathlib4 during the image build
WORKDIR /app/mathlib4 # WORKDIR /app/mathlib4
RUN lake build # RUN lake exe cache get && lake build
# Set default work directory back to project root # Set default work directory back to project root
WORKDIR /app WORKDIR /app
@ -37,3 +25,4 @@ WORKDIR /app
# EXPOSE 8080 # EXPOSE 8080
CMD ["bash"] CMD ["bash"]

4
requirements.txt
View File

@ -1,6 +1,6 @@
torch==2.2.1
pytz==2022.1 pytz==2022.1
easydict==1.13 easydict==1.13
torch==2.2.1
transformers==4.40.1 transformers==4.40.1
vllm==0.4.1 vllm==0.4.1
numpy==1.26.4 numpy==1.26.4
@ -8,4 +8,4 @@ pandas==1.4.3
tabulate==0.9.0 tabulate==0.9.0
termcolor==2.4.0 termcolor==2.4.0
accelerate==0.33.0 accelerate==0.33.0
flash_attn==2.6.3 flash_attn==2.6.3