Skip to content

In this quickstart, we'll explore how to harness the power of Snowpark Container Services SPCS using GPUs to execute a notebook to train an XGBoost model directly within Snowflake.

License

Notifications You must be signed in to change notification settings

Snowflake-Labs/sfguide-train-xgboost-model-using-gpus-using-snowflake-notebooks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Snowflake - Certified Solution

Train XGBoost using GPUs with Snowflake Notebooks

Overview

In this quickstart, we'll explore how to harness the power of Snowpark Container Services (SPCS) using GPUs to execute a notebook directly within Snowflake. Specifically, we'll train an XGBoost model and walk through a workflow that involves inspecting GPU resources, loading data from a Snowflake table, and setting up that data for modeling. In the notebook, we will train two XGBoost models—one on CPUs and the other using a GPU cluster—and then compare their runtimes and results.

Step-By-Step Guide

For prerequisites, environment setup, step-by-step guide and instructions, please refer to the QuickStart Guide.

About

In this quickstart, we'll explore how to harness the power of Snowpark Container Services SPCS using GPUs to execute a notebook to train an XGBoost model directly within Snowflake.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published