#25 - fix sm-local-configure for Linux: using sudo

Author: ivan-khvostishkov

kernel-lc-config.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-# A lifecycle configuration script for SageMaker Studio.
+# A lifecycle configuration script for SageMaker Studio kernel gateway.
 # See SageMaker_SSH_IDE.ipynb for manual configuration and for explanation of commands.
 # See https://docs.aws.amazon.com/sagemaker/latest/dg/studio-lcc.html .
 

sagemaker_ssh_helper/sm-helper-functions

@@ -39,16 +39,16 @@ function _is_macos() {
 
 function _install_unzip() {
   if _is_centos; then
-    yum install -y unzip
+    sudo yum install -y unzip
   else
-    apt-get install -y --no-install-recommends unzip
+    sudo apt-get install -y --no-install-recommends unzip
   fi
 }
 
 function _install_aws_cli_linux() {
   curl -sS "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "/tmp/awscliv2.zip"
   unzip -o -q -d /tmp/ /tmp/awscliv2.zip
-  /tmp/aws/install --update
+  sudo /tmp/aws/install --update
 }
 
 function _install_aws_cli_macos() {
@@ -92,11 +92,11 @@ function _install_ssm_agent() {
 function _install_session_manager_plugin_ubuntu() {
   curl -sS "https://s3.amazonaws.com/session-manager-downloads/plugin/latest/ubuntu_64bit/session-manager-plugin.deb" \
     -o "/tmp/session-manager-plugin.deb"
-  dpkg -i /tmp/session-manager-plugin.deb
+  sudo dpkg -i /tmp/session-manager-plugin.deb
 }
 
 function _install_session_manager_plugin_centos() {
-  yum install -y "https://s3.amazonaws.com/session-manager-downloads/plugin/latest/linux_64bit/session-manager-plugin.rpm" \
+  sudo yum install -y "https://s3.amazonaws.com/session-manager-downloads/plugin/latest/linux_64bit/session-manager-plugin.rpm" \
     || echo "Already installed?"
 }
 
@@ -123,11 +123,11 @@ function _install_session_manager_plugin() {
 
 
 function _install_curl_ubuntu() {
-  apt-get install -y --no-install-recommends curl
+  sudo apt-get install -y --no-install-recommends curl
 }
 
 function _install_curl_centos() {
-  yum install -y curl
+  sudo yum install -y curl
 }
 
 function _install_curl() {
@@ -153,3 +153,14 @@ function _install_sudo() {
     apt-get install -y --no-install-recommends sudo
   fi
 }
+
+function _install_noVNC_linux() {
+  json_value_regexp='s/^[^"]*".*": \"\(.*\)\"[^"]*/\1/'
+  latest_noVNC_release=$(curl -s https://api.github.com/repos/novnc/noVNC/releases/latest)
+
+  novnc_tag=$(echo "$latest_noVNC_release" | grep "tag_name" | sed -e "$json_value_regexp")
+  echo "Got latest noVNC release tag: $novnc_tag"
+
+  curl -sSL "https://api.github.com/repos/novnc/noVNC/tarball/$novnc_tag" -o "/tmp/noVNC.tgz"
+  tar xzf /tmp/noVNC.tgz --transform 's/^novnc-noVNC-[0-9a-f]\+/noVNC/'
+}

sagemaker_ssh_helper/sm-init-ssm

@@ -14,6 +14,7 @@ _install_helper_scripts
 
 set -e
 
+_install_sudo
 _install_unzip
 _install_curl
 _install_aws_cli
@@ -78,5 +79,4 @@ aid=$(echo $response | jq --raw-output '.ActivationId')
 echo Yes | amazon-ssm-agent -register -id "$aid" -code "$acode" -region "$CURRENT_REGION"
 
 # See https://docs.aws.amazon.com/systems-manager/latest/userguide/session-manager-getting-started-ssm-user-permissions.html
-_install_sudo
 echo "ssm-user ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/ssm-agent-users

sagemaker_ssh_helper/sm-local-configure

@@ -13,6 +13,8 @@ cat /etc/os-release || :
 python --version
 
 # Install AWS CLI
+_install_unzip
+_install_curl
 _install_aws_cli
 
 echo "AWS default region - $AWS_DEFAULT_REGION"

server-lc-config.sh

@@ -0,0 +1,35 @@
+#!/bin/bash
+
+# A lifecycle configuration script for SageMaker Studio Jupyter Server.
+# Prepares environment to use port forwarding into kernel gateways
+# through Jupyter Server Proxy extension and noVNC.
+# See https://docs.aws.amazon.com/sagemaker/latest/dg/studio-lcc.html .
+
+set -e
+
+sudo yum install -y hostname
+hostname
+cat /opt/ml/metadata/resource-metadata.json
+
+source /opt/conda/etc/profile.d/conda.sh
+conda activate base
+pip uninstall -y -q awscli
+pip install -q sagemaker-ssh-helper
+
+# Uncomment two lines below to update SageMaker SSH Helper to the latest dev version from main branch
+#git clone https://github.com/aws-samples/sagemaker-ssh-helper.git
+#cd sagemaker-ssh-helper && pip install . && cd ..
+
+ps xfaeww
+
+which python
+which pip
+
+env
+
+sm-local-configure
+
+_install_noVNC_linux
+
+cd noVNC
+nohup ./utils/novnc_proxy --vnc 127.0.0.1:5901 &