Google CoLab Enterprise at UMD


Table of contents

The intent of this article is to provide guidance in using Google CoLab Enterprise for students and researchers.

What is Colab Enterprise

Colab Enterprise is a collaborative notebook environment that's managed and offers the security and compliance capabilities of Google Cloud. It's a Vertex AI tool that combines the collaborative features of Colaboratory with Google Cloud's security and compliance capabilities. Colab Enterprise is intended for data analysts, data scientists, and data developers who work with notebook environments. 

The Google Cloud CoLab Enterprise environment is a UMD recommended fee based option instead of Google Workspace subscription solutions CoLab Pro and CoLab Pro+. UMD environment does not support credit card payments for Google Cloud Services. UMD CoLab Enterprise solution offers compatible features and instances to the subscription versions. Like the subscription option, CoLab Enterprise has costs associated with it. Contact DIT for more information.

Top

Begin using CoLab Enterprise at UMD

  1. Submit a Google Cloud Service Request Form.
  2. Generate using Google Cloud Pricing calculator an estimate of Google Cloud spending using CoLab Enterprise.
  3. Request a quote from DIT Research Computing team from estimate generated.
  4. Request a purchase order from your department business office against the quote generated.
  5. Once purchase order is generated, submit  copy of Purchase Order to DIT Research Computing.
  6. Google Cloud Project will be created against the service request form submitted by research computing.
  7. DIT Research Computing will notify you when project is ready for use.

Top

Colab Enterprise system configuration

Comparing Colab Enterprise

Here’s a comparison of the specifications and offerings for Colab, Colab Pro, and UMD Colab Enterprise. For more information, see the Colab GPUs Features & Pricing article.

Feature Google Colab Google Colab Pro UMD Colab Enterprise
GPUs T4 K80, P100, T4, A100 T4, V100,A100,* L4*
CPUs 2xvCPU 2xvCPU 4xvCPU to more
RAM 12GB 32GB 16GB - 60GB
Price Free 9.99/month for 100 compute units (~8-10 hrs total use per month) Dependent on usage
Datasets/Notebooks Upload/Download locations Local Drive, Google Drive, URL Local Drive, Google Drive, URL Local Drive, Google Cloud Storage Buckets, URL
Run/Code Execution Time Up to 12 hours Dependent on compute units usage Dependent on user and data requirements
Access Control Managed through Google Drive sharing credentials Subscription managed Managed through IAM UMD (user id and password)
Support Blogs, Internet   Standard Support provided for each project
Limitations System time outs due to reaching max GPU use Session time out when compute units needs have exceeded Can’t mount google drive within enterprise
  T4 is cheapest GPU Not approved for use within UMD Domain The file size of uploaded notebooks is limited to approximately 20 MB.

* Denotes specialized machine type required for high performance GPU.

Top

Colab Enterprise GPU availability regions

Google offers a number of virtual machines (VMs) that provide graphical processing units (GPUs), including the NVIDIA Tesla K80, P4, T4, P100, and V100.

You can use NVIDIA GPUs on GCP for large scale cloud deep learning projects, analytics, physical object simulation, video transcoding, and molecular modeling. GCP also provides virtual NVIDIA GRID workstations, which can let an organization’s employees run graphics-intensive workloads remotely.

Google Cloud’s Colab Enterprise utilizes 8 regions within the United States. The Colab enterprise service also limits the virtual machine types for GPU use to the N1, A2, G2 series.

Available machine types associated with possible GPU within the region offerings
Regions GPU      
South Carolina - us-east1 N1-Tesla T4  N1-Tesla V100 A2-Tesla V100 G2- NVIDIA-L4
Northern Va. - us-east4 N1-Tesla T4  G2- NVIDIA-L4 A2-Tesla V100  
Iowa - us-central1 N1-Tesla T4  N1-Tesla V100 A2-Tesla V100 G2- NVIDIA-L4
Oregon - us-west1 N1-Tesla T4  N1-Tesla V100 G2- NVIDIA-L4  
Los Angeles - us-west2 N1-Tesla T4  N1-Tesla V100    
Los Angeles - us-west4 N1-Tesla T4  N1-Tesla V100 A2-Tesla V100 G2- NVIDIA-L4

UMD recommended configurations for Colab Enterprise users

Regions for GPU use. The listed GPUs T4, V100, L4 and A100 are found in all three regions.  

  1. us-east1.
  2. us-central1.
  3. us-west4.

The below Virtual Machine types coincide with the listed regions for supported GPUs.  

  1. N1 and E2 without GPU.
  2. N1, G2, and A2 for GPU usage.

Top

Minimum roles and permissions for using Colab Enterprise

  1. Colab Enterprise User- ability to create, upload and share notebooks within project.
  2. Notebook Runtime User - ability to access and use runtime templates/runtimes within project.
  3. Storage Object User- ability to store datasets in Google Cloud Storage.
  4. Colab Enterprise Admin - Managing within the Colab Enterprise Project only.

Top

Colab Enterprise pricing assessment

During our assessment we were able to assess the cost to utilize a Colab enterprise project with a proposed configuration.

CoLab Pro prices vs CoLab Enterprise
GPU Computer units/hr Small models approx max runtime hours/month Google GPU cost/hr
T4 1.92 52 $.35 per GPU
L4 4.92 20 $2.48 per GPU
V100 5 20  
A100 15 7  

 

Cost breakdown example when using CoLab Enterprise
Google service CoLab Ent (no assigned runtime) CoLab Ent runtime (idle) (single runtime) Idle CoLab Ent (3 or more runtimes) CoLab Ent with T4 idle runtime CoLab Ent runtime with T4 (single GPU) large dataset
Vertex AI 0 $1.47/day $4.41/day $5.15/day $13.87/use
Compute Engine 0 $.011 0 0 $.04
Support 0 $.044 $.13 $.15 $1.4
Cloud Storage 0 0 0 0 $1.69
Total per use 0 $1.5/day $5/day   $17/use

The costs assessment from the table below that at first look using the subscription version will be the most cost effective solution. However, for the subscription to be the alternative the use cases must fall specifically within the cost variables. For example the use case utilizing a single GPU will allow a maximum of 52 hours when using the T4. This is ideal for small models generally needed for teaching and learning. However as the use case gets more complex, the subscription scenario is not as cost effective and Colab Enterprise is a more efficient solution, flexible and comparable in cost with the subscription versions.

Top

Tips when using Colab Enterprise

  1. If the project is no longer in use, Always delete the project once you no longer need it. There are costs associated with configured resources even inactive.
  2. To minimize costs for an active running project, delete runtime when you have completed running your analysis. Runtimes are considered reserved resources and Google will charge you a nominal fee ($.80 to $5/day) for it just sitting there. Cost is dependent on the number and types of runtime templates configured. Recommend to delete runtimes if it is a one time or once in a while use to minimize charges.
  3. Set the idle time before shutting down the resources to 30 min or under. Default runtimes are 180 min idle before shutting down resources. This can run your costs up in the long run.
  4. Regular users should only have the Notebook Runtime User role to prevent users from spinning up a default runtime configuration which is an automatic setting. With these permissions user’s are required to run only templates configured by administrator
  5. Uploading datasets can be done manually using Google Drive, local drive, UMD Box. If you choose to mount the drive within your code, mounting Google Drive is not supported to date.
  6. Save your analysis results frequently, save your data to local drive, UMD Box or Google Cloud Storage folders.

Top