FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime

RUN pip3 install numpy==1.21.2
RUN pip3 install scikit-learn==1.0.1
RUN pip3 install matplotlib==3.5.1
RUN pip3 install pandas==1.3.5
RUN pip3 install opencv-python-headless==4.5.5.62
RUN pip3 install albumentations==1.1.0
RUN pip3 install wandb==0.12.10
RUN pip3 install tqdm
RUN pip3 install protobuf==3.20.0
RUN pip3 install tensorboard_logger
RUN pip3 install gdown
RUN pip3 install tensorboard

RUN mkdir /.config
RUN mkdir /.cache
RUN mkdir /data
RUN mkdir /src
COPY . /src
WORKDIR /src

# Run with "docker run <image> main_infonce.py arg1 arg2 ...""
ENTRYPOINT ["python3"] 

