gpt-neo / Dockerfile
aliabd
full working demo
c6e7238
FROM gcr.io/deeplearning-platform-release/tf-cpu.1-15
WORKDIR /neogpt
# Make RUN commands use `bash --login`:
SHELL ["/bin/bash", "--login", "-c"]
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update -y && apt-get install tmux -y
RUN conda install gcc_linux-64 gxx_linux-64 -y
ADD requirements.txt .
RUN pip install -r requirements.txt
RUN apt-get install screen htop -y
RUN python -m pip install tensorboard==1.15 cloud_tpu_profiler==1.15
CMD tmux