ssh-Colab 0.3.0

Google Colab secure shell connection helper that automates ngrok tunnels creation (for SSH, TPU, and TensorBoard) and facilitates the use of Google Cloud Storage, Google Drive, and Kaggle Data API.

ssh-Colab

ssh-Colab is a Python module to facilitate remote access to Google Colaboratory (Colab) through Secure Shell (SSH) connections, secured by a third-party software, ngrok. The module automates the tedious routine to set up ngrok tunnels needed for TPU runtime applications and services like TensorBoard. It also provides subroutines for facilitating (1) Kaggle Data API installation, (2) Kaggle competition data downloads, (3) data transfers between Colab and Google Cloud Storage (GCS), and (4) Google Drive mounting.

Prerequisites

• ngrok tunnel authtoken.
• Google account to access a Colab notebook.
• Local code editors such as VS Code or PyCharm to make the most of coding on Colab.

Usage

1. Launch a Colab notebook. Choose a runtime type you prefer.

2. Install ssh-Colab. Type and run the following command in a notebook cell:

   !pip install ssh-Colab
   

3. Initiate the establishment of tunnels:

   import sshColab
   sshColab.connect([LOG_DIR='/path/to/log/'])
   

   The default TensorBoard log directory is /log/fit.

4. Retrieve information that is used for establishing the SSH connection:

   sshColab.info()
   

   If you are running a non-TPU-enabled notebook, the setup instruction of TPU resolver is skipped.

5. To activate Kaggle API installation/authentication and download competition data, run:

   sshColab.kaggle([data=<name-of-competition>])
   

   Note that the competition name is tabular-playground-series-mar-2021 by default. The data is unzipped to the destination folder /kaggle/input.

6. To mount a google drive, run:

7. To connect with GCS, initiate the connection:

   sshColab.GCSconnect()
   

   To create a GCS Bucket, run:

   sshColab.create_bucket(<project_id>, <bucket_name>)
   

   To list blobs in a GCS bucket, run:

   sshColab.list_blobs(<project_id>, <bucket_name>)
   

   To upload files from Colab to a GCS Bucket, run:

   sshColab.upload_to_gcs(<project_id>, <bucket_name>, [file=<local_file> ,ext=<file_extension>])
   

   To download files from a GCS Bucket to Colab, run:

   sshColab.download_to_colab(<project_id>, <bucket_name>, [file=<local_file>])
   

8. To disable ngrok tunnels created, run the command below:

   sshColab.kill()
   

Quickstart

A short Colab notebook is provided in the link below. Users can find a simple end-to-end application starting from ssh-Colab installation, SSH tunnel creation, to the use of TensorBoard after training a 3-layer MNIST convolutional neural network.

What's missed in this quick start guide is how to may our way to Colab instances from local machines. The reference listed below can be a start point for interested users:

1. Remote development over SSH on local VS Code
2. Run SSH terminal on local PyCharm

Releases

version 0.3.0: Addition of functions for communicating with Google Cloud Storage.

version 0.2.0: Addition of Google Drive mounting function.

version 0.1.3: Addition of Kaggle API installation/authentication and competition data downloading function.

Feedback

Comments and suggestions are welcome and appreciated. They can be sent to [e-mail].