Create Dockerfile

Dockerfile

@@ -0,0 +1,25 @@
+# This dockerfile is forked from ai2/cuda11.8-cudnn8-dev-ubuntu20.04
+FROM gcr.io/ai2-beaker-core/public/cjvktq5s0r0fr8pb7470:latest
+
+RUN apt update && apt install -y openjdk-8-jre-headless
+
+RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash
+RUN apt-get -y install git-lfs
+
+WORKDIR /stage/
+
+COPY requirements.txt .
+RUN pip install --upgrade pip setuptools wheel
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+RUN pip install packaging
+RUN pip install flash-attn --no-build-isolation
+RUN pip install -r requirements.txt
+
+COPY open_instruct open_instruct
+COPY eval eval
+COPY ds_configs ds_configs
+COPY scripts scripts
+RUN chmod +x scripts/*
+
+# for interactive session
+RUN chmod -R 777 /stage/