FROM ubuntu:20.04

RUN apt-get update
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get install -y git cmake build-essential libgl1-mesa-dev  libboost-all-dev \
libdirectfb-dev libst-dev mesa-utils xvfb x11vnc

RUN apt-get install -y libsm6 libxrender1 libxext-dev
RUN apt-get install -y libzmq3-dev

RUN apt-get install -y python3 python3-pip vim

RUN pip3 install scikit-build gym==0.16.0 numpy visdom lz4

RUN pip install pyzmq

WORKDIR /home
COPY torch-2.3.0+cpu-cp38-cp38-linux_x86_64.whl .
COPY torchvision-0.18.0+cpu-cp38-cp38-linux_x86_64.whl .
COPY torchaudio-2.3.0+cpu-cp38-cp38-linux_x86_64.whl .
RUN pip3 install torch-2.3.0+cpu-cp38-cp38-linux_x86_64.whl
RUN pip3 install torchvision-0.18.0+cpu-cp38-cp38-linux_x86_64.whl
RUN pip3 install torchaudio-2.3.0+cpu-cp38-cp38-linux_x86_64.whl
RUN pip3 install --upgrade pip setuptools psutil wheel flatbuffers
RUN rm -rf /home/torchvision-0.18.0+cpu-cp38-cp38-linux_x86_64.whl /home/torch-2.3.0+cpu-cp38-cp38-linux_x86_64.whl /home/torchaudio-2.3.0+cpu-cp38-cp38-linux_x86_64.whl
RUN apt-get update && apt-get install -y unzip

RUN pip3 install --upgrade pip setuptools psutil wheel
RUN pip3 install gymnasium==0.29.1 highway-env==1.8.2

RUN apt-get install -y wget
RUN mkdir /home/dragonfly
WORKDIR /home/dragonfly
COPY Dragonfly_1.0.2_linux_amd64.tar.gz .
RUN tar -zxvf Dragonfly_1.0.2_linux_amd64.tar.gz

COPY worker /home/gfootball/worker/
COPY config.json /home/gfootball/
COPY env_config.json /home/gfootball/
WORKDIR /home/gfootball/

RUN rm -rf /home/torchaudio-2.3.0+cpu-cp38-cp38-linux_x86_64.whl
RUN rm -rf /home/torch-2.3.0+cpu-cp38-cp38-linux_x86_64.whl

#################################################
