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

ColaBH · Aug 18, 2022 · d248e76 · d248e76
1 parent e14860d
commit d248e76
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/Jenkinsfile b/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
                 }

diff --git a/docker/Dockerfile.ci_cpu b/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=
+
+# 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
diff --git a/docker/install/ubuntu_install_core.sh b/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