.gitignore

@@ -1 +0,0 @@
-conda/

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04
+FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
 
 # encoding
 RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
@@ -10,19 +10,25 @@ RUN echo "LANG=en_US.UTF-8" > /etc/locale.conf
 ARG DEBIAN_FRONTEND=noninteractive
 
 
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
-
-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 apt-get -y update && \
+    apt-get -y upgrade && \
+    apt-get install -y python3-pip python3-dev
 
+RUN apt-get install -y locales git vim nano
+
+RUN apt-get install -y zsh && apt-get install -y curl
+RUN PATH="$PATH:/usr/bin/zsh"
+RUN sh -c "$(curl -fsSL https://raw.githubusercontent.com/robbyrussell/oh-my-zsh/master/tools/install.sh)"
 
 
 
+RUN echo 'alias python="python3"' >> ~/.bashrc
+RUN echo 'alias python="python3"' >> ~/.zshrc
+RUN echo 'alias pip="pip3"' >> ~/.bashrc
+RUN echo 'alias pip="pip3"' >> ~/.zshrc
 
 
+RUN apt-get install -y openssh-server
 RUN mkdir /var/run/sshd
 RUN echo 'root:qwerty' | chpasswd
 RUN sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
@@ -34,14 +40,4 @@ EXPOSE 22
 
 RUN usermod -s /bin/zsh root
 
-ENTRYPOINT service ssh start && 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
-
+ENTRYPOINT ["/bin/zsh"]