FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y libncurses5 python3 python3-pip curl git apt-utils
RUN pip3 install https://www.dropbox.com/s/51swz6j2vczh5u9/transformerfork-master.zip?dl=1 pika
RUN pip3 install prometheus_client PyNaCl websockets boto3 retry sentry-sdk icecream dotmap

EXPOSE 8080
EXPOSE 80
EXPOSE 4369
EXPOSE 5672
EXPOSE 25672
EXPOSE 15672
EXPOSE 15692

WORKDIR /usr/src/app
COPY . .
CMD [ "python3", "trainer_node.py" ]
