Dockerfile cloud ports (#1148)

* explicitly expose ports 8888 and 22

* support for SSH_KEY from latitude

docker/Dockerfile-cloud

@@ -7,6 +7,9 @@ ENV TRANSFORMERS_CACHE="/workspace/data/huggingface-cache/hub"
 ENV HF_HOME="/workspace/data/huggingface-cache/hub"
 ENV HF_HUB_ENABLE_HF_TRANSFER="1"
 
+EXPOSE 8888
+EXPOSE 22
+
 COPY scripts/cloud-entrypoint.sh /root/cloud-entrypoint.sh
 
 RUN pip install jupyterlab notebook && \

scripts/cloud-entrypoint.sh

@@ -7,14 +7,24 @@ echo 'source /etc/rp_environment' >> ~/.bashrc
 
 if [[ $PUBLIC_KEY ]]
 then
+    # runpod
     mkdir -p ~/.ssh
     chmod 700 ~/.ssh
     echo $PUBLIC_KEY >> ~/.ssh/authorized_keys
     chmod 700 -R ~/.ssh
     # Start the SSH service in the background
     service ssh start
+elif [[ $SSH_KEY ]]
+then
+    # latitude.sh
+    mkdir -p ~/.ssh
+    chmod 700 ~/.ssh
+    echo $SSH_KEY >> ~/.ssh/authorized_keys
+    chmod 700 -R ~/.ssh
+    # Start the SSH service in the background
+    service ssh start
 else
-    echo "No PUBLIC_KEY ENV variable provided, not starting openSSH daemon"
+    echo "No PUBLIC_KEY or SSH_KEY environment variable provided, not starting openSSH daemon"
 fi
 
 # Check if JUPYTER_PASSWORD is set and not empty