FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04

# encoding
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
RUN echo "en_US.UTF-8 UTF-8" >> /etc/locale.gen
RUN echo "LANG=en_US.UTF-8" > /etc/locale.conf


# fix (tzdata)
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update
RUN apt-get install -y locales git vim nano zsh wget openssh-server  && rm -rf /var/lib/apt/lists/*


RUN mkdir /var/run/sshd
RUN echo 'root:qwerty' | chpasswd
RUN sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
RUN sed -i 's/#PubkeyAuthentication yes/PubkeyAuthentication yes/' /etc/ssh/sshd_config
RUN sed 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' -i /etc/pam.d/sshd
ENV NOTVISIBLE "in users profile"
RUN echo "export VISIBLE=now" >> /etc/profile
EXPOSE 22

RUN service ssh start

RUN usermod -s /bin/zsh root

ENTRYPOINT ["/bin/zsh"]

COPY ./conda/Miniconda3-latest-Linux-x86_64.sh /opt/Miniconda3-latest-Linux-x86_64.sh

RUN bash /opt/Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda
RUN rm -f /opt/Miniconda3-latest-Linux-x86_64.sh 
ENV PATH /opt/conda/bin:$PATH