Skip to content

Commit

Permalink
add docker
Browse files Browse the repository at this point in the history
  • Loading branch information
xingyaoww committed Mar 14, 2024
1 parent d0b5d30 commit 39d43ed
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 0 deletions.
35 changes: 35 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
FROM ubuntu:20.04

# https://github.com/princeton-nlp/SWE-bench/issues/15#issuecomment-1815392192
RUN apt-get update && \
apt-get install -y bash gcc git jq wget && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*

RUN git config --global user.email "[email protected]"
RUN git config --global user.name "swebench"

RUN apt update && apt install -y build-essential

# Create new user
RUN useradd -ms /bin/bash swe-bench
USER swe-bench
WORKDIR /home/swe-bench

# Setup Conda
ENV PATH="/home/swe-bench/miniconda3/bin:${PATH}"
ARG PATH="/home/swe-bench/miniconda3/bin:${PATH}"
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh \
&& mkdir ~/.conda \
&& bash miniconda.sh -b \
&& rm -f miniconda.sh
RUN conda --version

# Setup SWE-Bench Env
COPY environment.yml .
RUN conda env create -f environment.yml

# Some missing packages
RUN pip install datasets python-dotenv gitpython

CMD ["/bin/bash"]
13 changes: 13 additions & 0 deletions harness/run_docker_interactive.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
#!/bin/bash

DOCKER_IMAGE=xingyaoww/swe-bench
WORK_DIR=`pwd`

docker run \
-it \
--rm \
--user $(id -u):$(id -g) \
--ipc=host --ulimit memlock=-1 --ulimit stack=67108864 \
-v $WORK_DIR:/swe-bench \
$DOCKER_IMAGE \
bash -c "cd /swe-bench && bash"

0 comments on commit 39d43ed

Please sign in to comment.