diff --git a/Dockerfile b/Dockerfile
index 3a7163f..85cc4cf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu18.04
 
 # encoding
 RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
@@ -43,4 +43,4 @@ EXPOSE 22
 
 RUN usermod -s /bin/zsh root
 
-ENTRYPOINT ["/bin/zsh"]
+ENTRYPOINT service ssh start && bash