[CI] enable ssh in docker image for dist test (#4432)

Jenkinsfile

@@ -464,7 +464,7 @@ pipeline {
               agent {
                 docker {
                   label "linux-cpu-node"
-                  image "dgllib/dgl-ci-cpu:v220816"
+                  image "dgllib/dgl-ci-cpu:ssh_v220818"
                   args "--shm-size=4gb"
                   alwaysPull true
                 }

docker/Dockerfile.ci_cpu

@@ -29,4 +29,9 @@ RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/tensorflow
 COPY install/conda_env/mxnet_cpu.yml /install/conda_env/mxnet_cpu.yml
 RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/mxnet_cpu.yml"]
 
-ENV CONDA_ALWAYS_YES=
+ENV CONDA_ALWAYS_YES=
\ No newline at end of file
+
+# SSH
+RUN ["/bin/bash", "-i", "-c", "ssh-keygen -f ~/.ssh/id_rsa -N ''"]
+RUN ["/bin/bash", "-i", "-c", "cat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys"]
+ENTRYPOINT service ssh restart && bash

docker/install/ubuntu_install_core.sh

@@ -2,4 +2,5 @@
 apt update && apt install -y --no-install-recommends --force-yes \
         apt-utils git build-essential make wget unzip sudo \
         libz-dev libxml2-dev libopenblas-dev libopencv-dev \
-        graphviz graphviz-dev libgraphviz-dev ca-certificates
+        graphviz graphviz-dev libgraphviz-dev ca-certificates \
+        systemd vim openssh-client openssh-server