FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime

WORKDIR /tmp

ENV DEBIAN_FRONTEND=noninteractive \
    LC_ALL=ko_KR.UTF-8 \
    TZ=Asia/Seoul \
    RUNZSH=no

RUN apt update && \
    apt install -y --no-install-recommends build-essential git zsh curl vim less locales locales-all python3-dev && \
    locale-gen ko_KR.UTF-8 && \
    rm -rf /var/lib/apt/lists/* && \
    sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" && \
    chsh -s $(which zsh) && \
    conda init zsh

RUN git clone https://github.com/zsh-users/zsh-syntax-highlighting.git \
    ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/zsh-syntax-highlighting && \
    git clone https://github.com/djui/alias-tips.git \
    ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/alias-tips && \
    perl -pi -e 's/plugins\=\(git\)/plugins\=\(git zsh-syntax-highlighting alias-tips\)/g' ~/.zshrc

COPY requirements.txt requirements.txt
RUN pip install --ignore-installed -r requirements.txt --no-cache-dir

ENV PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python

WORKDIR /workspace