FROM ubuntu | |
WORKDIR /app | |
RUN apt-get update && \ | |
apt-get upgrade -y && \ | |
apt-get install -y git && \ | |
apt-get install -y libopenmpi-dev && \ | |
apt-get install -y python3-pip && \ | |
git clone https://github.com/kitamoto-lab/pyphoon2.git && \ | |
cd pyphoon2 && \ | |
pip3 install . && \ | |
pip3 install tqdm && \ | |
pip3 install scikit-learn && \ | |
pip3 install matplotlib && \ | |
pip3 install seaborn && \ | |
pip3 install lightning && \ | |
pip3 install tensorboardX | |