FROM gooseai/torch-base:6cfdc11
RUN apt-get install -y cuda-nvcc-11-3 cuda-nvml-dev-11-3 libcurand-dev-11-3 \
                       libcublas-dev-11-3 libcusparse-dev-11-3 \
                       libcusolver-dev-11-3 cuda-nvprof-11-3 && \
    apt-get clean

#Open ports
EXPOSE 8080
EXPOSE 80
EXPOSE 443
EXPOSE 4369
EXPOSE 5672
EXPOSE 25672
EXPOSE 15672
EXPOSE 15692
EXPOSE 50051

RUN mkdir -p /usr/src/app
WORKDIR /usr/src/app
# Requirements
COPY requirements.txt .
# Copy our transformers fork in.
COPY transformers transformers
RUN pip3 install --no-cache-dir -r requirements.txt
# Then our hydra-node source
COPY . .
CMD [ "/usr/bin/python3", "main.py" ]
