Skip to navigation
Getting Started

Run Training Using Financial Fraud Training

View as Markdown

This document provides step-by-step instructions to log in, pull the Financial Fraud Training Docker image, run the Financial Fraud Training container, and train models using it.

1. Authenticate with NVIDIA GPU Cloud (NGC) and Pull the Docker Image

The following explains how to authenticate with the NVIDIA GPU Cloud (NGC) and pull the Docker image from the NVIDIA registry.

Prerequisites

  • NGC Account & API Key: Ensure you have an NGC account and a valid API key.
  • Docker: Make sure Docker is installed and running on your machine.

Authenticate with NGC

Log in to the NVIDIA container registry using your NGC API key:

docker login nvcr.io --username '$oauthtoken' --password $NGC_API_KEY
Replace $NGC_API_KEY with your actual API key or set it in NGC_API_KEY env variable.

Pull the Docker Image

After authenticating, pull the desired Docker image from the registry:

docker pull nvcr.io/nvidia/cugraph/financial-fraud-training:2.0.0

2. Train models

Run the Container

Run the training container with the following command:

    docker run -it --rm --name=financial-fraud-training \
    --gpus "device=0" -v <YOUR_GNN_DATA_DIR>:/data \
    -v <DIR_TO_SAVE_TRAINED_MODELS>:/trained_models \
    nvcr.io/nvidia/cugraph/financial-fraud-training:2.0.0

Command Breakdown

Docker Run Options:

-it: Run the container interactively.

—rm: Automatically removes the container when it exits.

—name=financial-fraud-training: Names the container.

Volume Mounts:

-v <YOUR_GNN_DATA_DIR>:/data: Mounts your GNN data directory into the container.

-v <DIR_TO_SAVE_TRAINED_MODELS>:/trained_models: Mounts a directory for saving trained models.

Replace <YOUR_GNN_DATA_DIR> and <DIR_TO_SAVE_TRAINED_MODELS> with the actual paths on your system where your GNN data is located and trained models should reside.