FROM pytorch/pytorch:latest
MAINTAINER ANONYMOUS_USER
RUN useradd -u xxx -ms /bin/bash xxx

# Run upgrades
RUN apt-get update
RUN apt-get install -y vim libopenmpi-dev wget unzip

# Install basic packages
RUN pip install cma==2.2.0 mpi4py pillow
RUN pip install tensorboardX
RUN pip install opencv-python
RUN apt update && apt install -y libsm6 libxext6
RUN pip install matplotlib
RUN pip install sklearn
RUN pip install seaborn
RUN pip install bayesian-optimization
