kikagaku/ml repository overview

FROM nvcr.io/nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04

RUN apt update -y
RUN apt-get update -y
RUN apt upgrade -y
RUN apt install -y wget build-essential gcc zlib1g-dev libssl-dev libhdf5-dev
RUN apt install -y sudo wget curl vim git

# Python
RUN apt install -y python3-pip

# Standard packages
RUN pip3 install -U pip
RUN pip3 install numpy==1.16.4
RUN pip3 install scipy sklearn pandas matplotlib seaborn plotly jupyter pillow

# OpenCV
ENV TZ='Asia/Tokyo'
ENV DEBIAN_FRONTEND=noninteractive
RUN apt install -y libopencv-dev
RUN pip3 install opencv-python
RUN pip3 install opencv-contrib-python

# MeCab
RUN cd /tmp \
    && wget "https://drive.google.com/uc?export=download&id=0B4y35FiV1wh7cENtOXlicTFaRUE" -O mecab.tar.gz \
    && tar zxvf mecab.tar.gz \
    && cd mecab-0.996 \
    && ./configure \
    && make \
    && make install
RUN pip install mecab-python3
RUN ldconfig
RUN cd /tmp \
    && git clone --depth 1 https://github.com/neologd/mecab-ipadic-neologd.git \
    && cd mecab-ipadic-neologd \
    && ./bin/install-mecab-ipadic-neologd -n -y
COPY mecabrc /usr/local/etc/mecabrc
RUN echo 'export LD_LIBRARY_PATH="/usr/local/lib:$LD_LIBRARY_PATH"' >> ~/.bash_profile

RUN pip install cupy-cuda100
RUN pip3 install chainer==6.4
RUN pip3 install torch==1.3
RUN pip3 install torchvision pytorch-lightning
RUN pip3 install tensorflow-gpu==2.0.0

RUN pip3 install plotly
RUN pip3 install pydot
RUN apt install graphviz -y
RUN pip3 install graphviz
RUN pip3 install -U pandas_datareader
RUN pip3 install pystan fbprophet

RUN pip3 install ax-platform

RUN pip3 install lightgbm
RUN pip3 install xgboost

# Jupyter Notebook
RUN jupyter notebook --generate-config
COPY jupyter_notebook_config.py /root/.jupyter/jupyter_notebook_config.py

# Execution command at startup
CMD cd /home && jupyter notebook