diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..9b89f28
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+conda/
diff --git a/Dockerfile b/Dockerfile
index 95799ae..fb7e2af 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,37 +1,18 @@
-FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
+FROM nvcr.io/nvidia/cuda:10.1-cudnn7-devel-ubuntu18.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-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
-RUN apt-get -y update && \
-    apt-get -y upgrade && \
-    apt-get install -y python3.8 python3-pip
-ENV PYTHON_VERSION=3.8
-
-RUN apt-get install -y locales git vim nano
-
-RUN apt-get install -y zsh && apt-get install -y curl
-ENV PATH="$PATH:/usr/bin/zsh"
-ENV PATH=/usr/bin:$PATH
-RUN sh -c "$(curl -fsSL https://raw.githubusercontent.com/robbyrussell/oh-my-zsh/master/tools/install.sh)"
+RUN apt-get update
+RUN apt-get install -y locales git vim nano zsh wget openssh-server  && rm -rf /var/lib/apt/lists/*
 
 
 
-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
@@ -41,6 +22,16 @@ ENV NOTVISIBLE "in users profile"
 RUN echo "export VISIBLE=now" >> /etc/profile
 EXPOSE 22
 
+
 RUN usermod -s /bin/zsh root
 
-ENTRYPOINT service ssh start && bash
+ENTRYPOINT service ssh start && zsh
+
+
+
+
+
+#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
+
+